unet-pytorch
There are 201 repositories under unet-pytorch topic.
qubvel-org/segmentation_models.pytorch
Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.
wolny/pytorch-3dunet
3D U-Net model for volumetric semantic segmentation written in pytorch
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
rishikksh20/ResUnet
Pytorch implementation of ResUnet and ResUnet ++
neptune-ai/open-solution-mapping-challenge
Open solution to the Mapping Challenge :earth_americas:
NITR098/Awesome-U-Net
Official repo for Medical Image Segmentation Review: The Success of U-Net
stefanknegt/Probabilistic-Unet-Pytorch
A Probabilistic U-Net for segmentation of ambiguous images implemented in PyTorch
sdsubhajitdas/Brain-Tumor-Segmentation
Brain Tumor Segmentation done using U-Net Architecture.
tugstugi/pytorch-saltnet
Kaggle | 9th place single model solution for TGS Salt Identification Challenge
HansBambel/SmaAt-UNet
PyTorch-Code for the Paper "SmaAt-UNet: Precipitation Nowcasting using a Small, Attentive UNet-Architecture"
pykao/Modified-3D-UNet-Pytorch
This repository implements pytorch version of the modifed 3D U-Net from Fabian Isensee et al. participating in BraTS2017
fepegar/unet
"pip install unet": PyTorch Implementation of 1D, 2D and 3D U-Net architecture.
zhoudaxia233/EfficientUnet-PyTorch
A PyTorch 1.0 Implementation of Unet with EfficientNet as encoder
minerva-ml/open-solution-data-science-bowl-2018
Open solution to the Data Science Bowl 2018
musimab/Tc_ID_Card_OCR
Automatic information extraction from identity card with ocr
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.
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.
Confusezius/unet-lits-2d-pipeline
Liver Lesion Segmentation with 2D Unets
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.
Diyago/Severstal-Steel-Defect-Detection
Can you detect and classify defects in steel? Segmentation in Pytorch
Allenem/MedicalImage
:hospital: 医学图像作业：图像配准论文阅读；眼底血管分割实验
anishreddy3/Crack-Semantic-Segmentation
Real time crack segmentation using PyTorch, OpenCV and ONNX runtime
France1/unet-multiclass-pytorch
Multiclass semantic segmentation using U-Net architecture combined with strong image augmentation
HowieMa/lstm_multi_modal_UNet
[ICIVC 2019] "LSTM multi-modal UNet for Brain Tumor Segmentation"
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
josedolz/IVD-Net
Repository containing the source code of the IVD-Net segmentation network that we proposed for the MICCAI 2018 IVD segmentation challenge.
zdaiot/Kaggle-Steel-Defect-Detection
My solution to the Severstal: Steel Defect Detection on Kaggle, which got the 96th place. (Top4%)
Thvnvtos/Lung_Segmentation
Lung Segmentation UNet model on 3D CT scans
DevJake/EEG-diffusion-pytorch
An implementation of Denoising Diffusion Probabilistic Models (DDPM) in PyTorch for EEG-based image reconstruction.
aws-samples/deploy-stable-diffusion-model-on-amazon-sagemaker-endpoint
Deploy Stable Diffusion Model on Amazon SageMaker Endpont
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.
AImageLab-zip/alveolar_canal
This repository contains the material from the paper "Improving Segmentation of the Inferior Alveolar Nerve through Deep Label Propagation"
Koldim2001/Unet-pytorch-training
Репозиторий для обучения нейросетевых моделей по семантической сегментации + пример использования моделей на практике
FengheTan9/Mobile-U-ViT
[ACM MM 2025] Mobile U-ViT: Revisiting large kernel and U-shaped ViT for efficient medical image segmentation
yakhyo/crack-segmentation
🛣️ | Road crack segmentation using UNet in PyTorch > Implementation of different loss functions (i.e Focal, Dice, Dice + CE)