Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.
-
Updated
Mar 10, 2025 - Python
Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.
Pytorch implementation of ResUnet and ResUnet ++
Open solution to the Mapping Challenge 🌎
A Probabilistic U-Net for segmentation of ambiguous images implemented in PyTorch
Kaggle | 9th place single model solution for TGS Salt Identification Challenge
PyTorch-Code for the Paper "SmaAt-UNet: Precipitation Nowcasting using a Small, Attentive UNet-Architecture"
This repository implements pytorch version of the modifed 3D U-Net from Fabian Isensee et al. participating in BraTS2017
This custom_node for ComfyUI adds one-click "Virtual VRAM" for any GGUF UNet and CLIP loader, managing the offload 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. Includes 16 examples covering common use cases.
"pip install unet": PyTorch Implementation of 1D, 2D and 3D U-Net architecture.
A PyTorch 1.0 Implementation of Unet with EfficientNet as encoder
Open solution to the Data Science Bowl 2018
Segmentation deep learning ALgorithm based on MONai toolbox: single and multi-label segmentation software developed by QIMP team-Vienna.
Automatic information extraction from identity card with ocr
Liver Lesion Segmentation with 2D Unets
This repository contains the code for semantic segmentation of the retina blood vessel on the DRIVE dataset using the PyTorch framework.
Real time crack segmentation using PyTorch, OpenCV and ONNX runtime
Can you detect and classify defects in steel? Segmentation in Pytorch
[ICIVC 2019] "LSTM multi-modal UNet for Brain Tumor Segmentation"
PyTorch implementation for Semantic Segmentation, include FCN, U-Net, SegNet, GCN, PSPNet, Deeplabv3, Deeplabv3+, Mask R-CNN, DUC, GoogleNet, and more dataset
🏥 医学图像作业:图像配准论文阅读;眼底血管分割实验
Add a description, image, and links to the unet-pytorch topic page so that developers can more easily learn about it.
To associate your repository with the unet-pytorch topic, visit your repo's landing page and select "manage topics."