unet-pytorch

There are 2 repositories under unet-pytorch topic.

qubvel-org / segmentation_models.pytorch
Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.
segmentation image-processing pspnet unet unet-pytorch pytorch fpn models imagenet semantic-segmentation image-segmentation segmentation-models unetplusplus deeplabv3 deeplab-v3-plus pretrained-weights computer-vision segformer transformers dpt
Language:Python 11037
wolny / pytorch-3dunet
3D U-Net model for volumetric semantic segmentation written in pytorch
3d-segmentation 3d-unet dice-coefficient groupnorm pytorch pytorch-3dunet residual-unet semantic-segmentation unet unet-pytorch volumetric-data
Language:Jupyter Notebook 2294
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
comfyui comfyui-nodes comfyui-workflow pytorch stable-diffusion ggml gguf-models unet-pytorch wanvideowrapper
Language:Python 604
rishikksh20 / ResUnet
Pytorch implementation of ResUnet and ResUnet ++
resunet unet image-processing image-segmentation unet-pytorch medical-image-processing satellite-imagery deep-residual-network segmentation image-to-image-translation
Language:Python 524
neptune-ai / open-solution-mapping-challenge
Open solution to the Mapping Challenge :earth_americas:
data-science machine-learning deep-learning kaggle python satellite-imagery data-science-learning lightgbm unet unet-image-segmentation unet-pytorch neptune machine-learning-lab mapping-challenge crowdai competition pipeline pipeline-framework
Language:Python 388
NITR098 / Awesome-U-Net
Official repo for Medical Image Segmentation Review: The Success of U-Net
medical-image-processing medical-image-segmentation segmentation-models transformer u-net unet-image-segmentation unet-pytorch
Language:Jupyter Notebook 350
stefanknegt / Probabilistic-Unet-Pytorch
A Probabilistic U-Net for segmentation of ambiguous images implemented in PyTorch
unet-image-segmentation unet-pytorch
Language:Python 290
sdsubhajitdas / Brain-Tumor-Segmentation
Brain Tumor Segmentation done using U-Net Architecture.
brain-tumor-segmentation deep-learning image-segmentation pytorch tumor-segmentation unet unet-image-segmentation unet-pytorch
Language:Jupyter Notebook 287
tugstugi / pytorch-saltnet
Kaggle | 9th place single model solution for TGS Salt Identification Challenge
deep-learning cnn pytorch segmentation convolutional-neural-networks unet unet-image-segmentation unet-pytorch seismic-imaging kaggle-competition
Language:Python 281
HansBambel / SmaAt-UNet
PyTorch-Code for the Paper "SmaAt-UNet: Precipitation Nowcasting using a Small, Attentive UNet-Architecture"
weather-forecast unet convolutional-neural-networks unet-pytorch precipitation-nowcasting
Language:Python 280
pykao / Modified-3D-UNet-Pytorch
This repository implements pytorch version of the modifed 3D U-Net from Fabian Isensee et al. participating in BraTS2017
3d-unet unet unet-3d unet-image-segmentation unet-pytorch
Language:Python 222
fepegar / unet
"pip install unet": PyTorch Implementation of 1D, 2D and 3D U-Net architecture.
unet-image-segmentation unet unet-pytorch deep-learning convolutional-neural-networks medical-imaging
Language:Python 179
zhoudaxia233 / EfficientUnet-PyTorch
A PyTorch 1.0 Implementation of Unet with EfficientNet as encoder
unet unet-pytorch image-segmentation efficientnet efficientunet
Language:Python 163
minerva-ml / open-solution-data-science-bowl-2018
Open solution to the Data Science Bowl 2018
data-science data-science-bowl data-science-bowl-2018 deep-learning image-processing kaggle machine-learning medical-image-processing medical-imaging neptune open-source python python3 segmentation unet unet-image-segmentation unet-pytorch
Language:Python 156
musimab / Tc_ID_Card_OCR
Automatic information extraction from identity card with ocr
ocr craft unet-pytorch easyocr id-card-recognition face-recognition computer-vision
Language:Python 126
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.
neu-det defect-detection defect-prediction segmentation-models unet-image-segmentation unet-pytorch resnet resnet-18 pytorch deep-learning deep-neural-networks
Language:Jupyter Notebook 124
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.
segmentation monai organs pytorch unet-pytorch unet unet-3d deep-learning monai-toolbox nnunet transformers unetr mri
Language:Python 123
Confusezius / unet-lits-2d-pipeline
Liver Lesion Segmentation with 2D Unets
deep-learning medical-image-segmentation unet-pytorch lesion-segmentation liver-segmentation unet semantic-segmentation
Language:Python 99
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.
drive-dataset image-segmentation pytorch retinal-vessel-segmentation unet unet-pytorch unet-segmentation
Language:Python 71
Diyago / Severstal-Steel-Defect-Detection
Can you detect and classify defects in steel? Segmentation in Pytorch
deep-learning unet-pytorch unet pytorch python
Language:Python 67
Allenem / MedicalImage
:hospital: 医学图像作业：图像配准论文阅读；眼底血管分割实验
medical-imaging registration segementation unet unet-pytorch unetplusplus
Language:Python 66
anishreddy3 / Crack-Semantic-Segmentation
Real time crack segmentation using PyTorch, OpenCV and ONNX runtime
cnn computer-vision crack-detection deep-learning image-processing neural-networks onnx opencv pytorch unet-pytorch
Language:Python 64
France1 / unet-multiclass-pytorch
Multiclass semantic segmentation using U-Net architecture combined with strong image augmentation
biological-image-processing computer-vision data-augmentation deep-learning multiclass-image-classification pytorch unet-image-segmentation unet-pytorch
Language:Jupyter Notebook 61
HowieMa / lstm_multi_modal_UNet
[ICIVC 2019] "LSTM multi-modal UNet for Brain Tumor Segmentation"
multi-modal-imaging medical-imaging brats-dataset unet-pytorch lstm-cnn
Language:Python 59
Charmve / Semantic-Segmentation-PyTorch
PyTorch implementation for Semantic Segmentation, include FCN, U-Net, SegNet, GCN, PSPNet, Deeplabv3, Deeplabv3+, Mask R-CNN, DUC, GoogleNet, and more dataset
semantic-segmentation pytorch pytorch-implementation computer-vision image-processing image-classification segnet unet-pytorch mask-rcnn fcn deep-learning charmve
Language:Python 56
josedolz / IVD-Net
Repository containing the source code of the IVD-Net segmentation network that we proposed for the MICCAI 2018 IVD segmentation challenge.
deep-learning medical-imaging medical-image-processing image-segmentation segmentation unet unet-image-segmentation unet-pytorch fcn multi-modal-imaging densenet hyperdensenet miccai pytorch multi-modal-unet intervertebral-disc-localization ivd-net pytorch-cnn
Language:Python 53
zdaiot / Kaggle-Steel-Defect-Detection
My solution to the Severstal: Steel Defect Detection on Kaggle, which got the 96th place. (Top4%)
kaggle unet-pytorch segmentation
Language:Python 53
Thvnvtos / Lung_Segmentation
Lung Segmentation UNet model on 3D CT scans
unet unet-pytorch lung-segmentation segmentation deep-learning
Language:Python 51
DevJake / EEG-diffusion-pytorch
An implementation of Denoising Diffusion Probabilistic Models (DDPM) in PyTorch for EEG-based image reconstruction.
bci brain-computer-interface ddpm deep-learning denoising-diffusion-models eeg electroencephalography generative-model image-reconstruction machine-learning pytorch unet unet-pytorch
Language:Jupyter Notebook 47
lachinov / brats2019
segmentation brats brats2018 brats2019 unet unet-pytorch unet-3d brain-tumor-segmentation
Language:Jupyter Notebook 46
deploy-stable-diffusion-model-on-amazon-sagemaker-endpoint
aws-samples / deploy-stable-diffusion-model-on-amazon-sagemaker-endpoint
Deploy Stable Diffusion Model on Amazon SageMaker Endpont
clip huggingface-transformers-pipeline inference multimodality pytorch sagemaker-deployment sagemaker-pytorch stablediffusion unet-pytorch vae
Language:Jupyter Notebook 38
chathumal93 / Pytorch-UNet-Flood-Segmentation
This repository contains a PyTorch implementation of a U-Net model for segmenting water areas (flood and permananet water) in Sentinel-1 satellite images.
sar sentinel-1 unet unet-pytorch unet-segmentation google-earth-engine
Language:Jupyter Notebook 36
alveolar_canal
AImageLab-zip / alveolar_canal
This repository contains the material from the paper "Improving Segmentation of the Inferior Alveolar Nerve through Deep Label Propagation"
segmentation-network unet-pytorch inferior-alveolar-canal inferior-alveolar-nerve
Language:Python 35
Koldim2001 / Unet-pytorch-training
Репозиторий для обучения нейросетевых моделей по семантической сегментации + пример использования моделей на практике
camvid-dataset cvat hair-segmentation pytorch-segmentation segmentation segmentation-training semantic-segmentation skin-segmentation streamlit-webapp unet unet-image-segmentation unet-pytorch
Language:Jupyter Notebook 35
FengheTan9 / Mobile-U-ViT
[ACM MM 2025] Mobile U-ViT: Revisiting large kernel and U-shaped ViT for efficient medical image segmentation
lightweight medical-image-segmentation segmentation transformer u-net unet unet-pytorch
Language:Python 32
yakhyo / crack-segmentation
🛣️ | Road crack segmentation using UNet in PyTorch > Implementation of different loss functions (i.e Focal, Dice, Dice + CE)
crack-segmentation dice-loss focal-loss pytorch-segmentation road-crack-segmentation segmentation unet-pytorch
Language:Jupyter Notebook 31

unet-pytorch

qubvel-org / segmentation_models.pytorch

wolny / pytorch-3dunet

pollockjj / ComfyUI-MultiGPU

rishikksh20 / ResUnet

neptune-ai / open-solution-mapping-challenge

NITR098 / Awesome-U-Net

stefanknegt / Probabilistic-Unet-Pytorch

sdsubhajitdas / Brain-Tumor-Segmentation

tugstugi / pytorch-saltnet

HansBambel / SmaAt-UNet

pykao / Modified-3D-UNet-Pytorch

fepegar / unet

zhoudaxia233 / EfficientUnet-PyTorch

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

musimab / Tc_ID_Card_OCR

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

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

Confusezius / unet-lits-2d-pipeline

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

Diyago / Severstal-Steel-Defect-Detection

Allenem / MedicalImage

anishreddy3 / Crack-Semantic-Segmentation

France1 / unet-multiclass-pytorch

HowieMa / lstm_multi_modal_UNet

Charmve / Semantic-Segmentation-PyTorch

josedolz / IVD-Net

zdaiot / Kaggle-Steel-Defect-Detection

Thvnvtos / Lung_Segmentation

DevJake / EEG-diffusion-pytorch

lachinov / brats2019

aws-samples / deploy-stable-diffusion-model-on-amazon-sagemaker-endpoint

chathumal93 / Pytorch-UNet-Flood-Segmentation

AImageLab-zip / alveolar_canal

Koldim2001 / Unet-pytorch-training

FengheTan9 / Mobile-U-ViT

yakhyo / crack-segmentation