#

unet-pytorch

Here are 115 public repositories matching this topic...

qubvel-org / segmentation_models.pytorch

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

Updated Dec 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

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

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

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

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

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

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 Oct 16, 2025
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

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

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

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

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

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

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

anishreddy3 / Crack-Semantic-Segmentation

Real time crack segmentation using PyTorch, OpenCV and ONNX runtime

opencv computer-vision deep-learning image-processing cnn pytorch neural-networks crack-detection onnx unet-pytorch

Updated Jul 16, 2022
Python

HowieMa / lstm_multi_modal_UNet

[ICIVC 2019] "LSTM multi-modal UNet for Brain Tumor Segmentation"

medical-imaging lstm-cnn unet-pytorch multi-modal-imaging brats-dataset

Updated Mar 14, 2020
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

zdaiot / Kaggle-Steel-Defect-Detection

My solution to the Severstal: Steel Defect Detection on Kaggle, which got the 96th place. (Top4%)

kaggle segmentation unet-pytorch

Updated Jun 27, 2023
Python

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 pytorch medical-imaging segmentation densenet fcn image-segmentation unet medical-image-processing miccai pytorch-cnn hyperdensenet unet-pytorch unet-image-segmentation multi-modal-imaging multi-modal-unet intervertebral-disc-localization ivd-net

Updated Dec 14, 2018
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."