Skip to content

#

unet-pytorch

Here are 195 public repositories matching this topic...

Language: All

Filter by language

All 195 Python 110 Jupyter Notebook 77 HTML 2 TeX 2 C++ 1

Sort: Most stars

Sort options

Most stars Fewest stars Most forks Fewest forks Recently updated Least recently updated

qubvel-org / segmentation_models.pytorch

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.

computer-vision models image-processing transformers pytorch imagenet segmentation image-segmentation unet dpt semantic-segmentation pretrained-weights pspnet fpn deeplabv3 unet-pytorch deeplab-v3-plus segmentation-models unetplusplus segformer

Updated Aug 29, 2025
Python

wolny / pytorch-3dunet

3D U-Net model for volumetric semantic segmentation written in pytorch

pytorch unet semantic-segmentation volumetric-data 3d-segmentation dice-coefficient unet-pytorch groupnorm 3d-unet pytorch-3dunet residual-unet

Updated Feb 27, 2025
Jupyter Notebook

rishikksh20 / ResUnet

Pytorch implementation of ResUnet and ResUnet ++

image-processing segmentation satellite-imagery image-segmentation unet deep-residual-network medical-image-processing image-to-image-translation unet-pytorch resunet

Updated Sep 7, 2024
Python

pollockjj / ComfyUI-MultiGPU

This custom_node for ComfyUI adds one-click "Virtual VRAM" for any UNet and CLIP loader as well MultiGPU integration in WanVideoWrapper, managing the offload/Block Swap of layers to DRAM *or* VRAM to maximize the latent space of your card. Also includes nodes for directly loading entire components (UNet, CLIP, VAE) onto the device you choose

pytorch unet-pytorch stable-diffusion comfyui ggml comfyui-workflow comfyui-nodes gguf-models wanvideowrapper

Updated Sep 1, 2025
Python

neptune-ai / open-solution-mapping-challenge

Open solution to the Mapping Challenge 🌎

python competition data-science machine-learning deep-learning pipeline neptune pipeline-framework kaggle lightgbm satellite-imagery unet data-science-learning crowdai unet-pytorch unet-image-segmentation machine-learning-lab mapping-challenge

Updated Mar 22, 2021
Python

NITR098 / Awesome-U-Net

Official repo for Medical Image Segmentation Review: The Success of U-Net

transformer medical-image-processing u-net medical-image-segmentation unet-pytorch unet-image-segmentation segmentation-models

Updated Sep 30, 2024
Jupyter Notebook

stefanknegt / Probabilistic-Unet-Pytorch

A Probabilistic U-Net for segmentation of ambiguous images implemented in PyTorch

unet-pytorch unet-image-segmentation

Updated Jan 16, 2019
Python

sdsubhajitdas / Brain-Tumor-Segmentation

Brain Tumor Segmentation done using U-Net Architecture.

deep-learning pytorch image-segmentation unet brain-tumor-segmentation unet-pytorch unet-image-segmentation tumor-segmentation

Updated Jul 21, 2023
Jupyter Notebook

tugstugi / pytorch-saltnet

Kaggle | 9th place single model solution for TGS Salt Identification Challenge

deep-learning cnn pytorch kaggle-competition segmentation convolutional-neural-networks unet seismic-imaging unet-pytorch unet-image-segmentation

Updated Oct 29, 2018
Python

HansBambel / SmaAt-UNet

PyTorch-Code for the Paper "SmaAt-UNet: Precipitation Nowcasting using a Small, Attentive UNet-Architecture"

convolutional-neural-networks weather-forecast unet unet-pytorch precipitation-nowcasting

Updated Mar 13, 2025
Python

pykao / Modified-3D-UNet-Pytorch

This repository implements pytorch version of the modifed 3D U-Net from Fabian Isensee et al. participating in BraTS2017

unet unet-3d unet-pytorch unet-image-segmentation 3d-unet

Updated Sep 7, 2018
Python

fepegar / unet

"pip install unet": PyTorch Implementation of 1D, 2D and 3D U-Net architecture.

deep-learning medical-imaging convolutional-neural-networks unet unet-pytorch unet-image-segmentation

Updated Dec 13, 2024
Python

zhoudaxia233 / EfficientUnet-PyTorch

A PyTorch 1.0 Implementation of Unet with EfficientNet as encoder

image-segmentation unet unet-pytorch efficientnet efficientunet

Updated Mar 13, 2021
Python

minerva-ml / open-solution-data-science-bowl-2018

Open solution to the Data Science Bowl 2018

python open-source data-science machine-learning deep-learning neptune image-processing python3 kaggle medical-imaging segmentation unet medical-image-processing data-science-bowl data-science-bowl-2018 unet-pytorch unet-image-segmentation

Updated Jun 22, 2022
Python

davidiommi / Pytorch--3D-Medical-Images-Segmentation--SALMON

Segmentation deep learning ALgorithm based on MONai toolbox: single and multi-label segmentation software developed by QIMP team-Vienna.

deep-learning transformers pytorch mri segmentation unet unet-3d unet-pytorch organs monai monai-toolbox nnunet unetr

Updated Nov 15, 2021
Python

NEU-DET-Steel-Surface-Defect-Detection

siddhartamukherjee / NEU-DET-Steel-Surface-Defect-Detection

This project is about detecting defects on steel surface using Unet. The dataset used for this project is the NEU-DET database.

deep-neural-networks deep-learning pytorch resnet defect-prediction resnet-18 defect-detection unet-pytorch unet-image-segmentation segmentation-models neu-det

Updated May 22, 2021
Jupyter Notebook

musimab / Tc_ID_Card_OCR

Automatic information extraction from identity card with ocr

ocr computer-vision craft face-recognition unet-pytorch id-card-recognition easyocr

Updated May 16, 2024
Python

Confusezius / unet-lits-2d-pipeline

Liver Lesion Segmentation with 2D Unets

deep-learning unet semantic-segmentation liver-segmentation medical-image-segmentation unet-pytorch lesion-segmentation

Updated Jul 4, 2019
Python

nikhilroxtomar / Retina-Blood-Vessel-Segmentation-in-PyTorch

This repository contains the code for semantic segmentation of the retina blood vessel on the DRIVE dataset using the PyTorch framework.

pytorch image-segmentation unet unet-pytorch retinal-vessel-segmentation unet-segmentation drive-dataset

Updated Jul 12, 2021
Python

Diyago / Severstal-Steel-Defect-Detection

Can you detect and classify defects in steel? Segmentation in Pytorch

python deep-learning pytorch unet unet-pytorch

Updated Sep 18, 2020
Python

Improve this page

Add a description, image, and links to the unet-pytorch topic page so that developers can more easily learn about it.

Curate this topic

Add this topic to your repo

To associate your repository with the unet-pytorch topic, visit your repo's landing page and select "manage topics."