unet-pytorch
Here are 201 public repositories matching this topic...
Sort:Most stars
Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.
- Updated
Oct 8, 2025 - Python
3D U-Net model for volumetric semantic segmentation written in pytorch
- Updated
Oct 6, 2025 - Jupyter Notebook
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
- Updated
Oct 7, 2025 - Python
Pytorch implementation of ResUnet and ResUnet ++
- Updated
Sep 7, 2024 - Python
Open solution to the Mapping Challenge 🌎
- Updated
Mar 22, 2021 - Python
Official repo for Medical Image Segmentation Review: The Success of U-Net
- Updated
Sep 30, 2024 - Jupyter Notebook
A Probabilistic U-Net for segmentation of ambiguous images implemented in PyTorch
- Updated
Jan 16, 2019 - Python
Brain Tumor Segmentation done using U-Net Architecture.
- Updated
Jul 21, 2023 - Jupyter Notebook
Kaggle | 9th place single model solution for TGS Salt Identification Challenge
- Updated
Oct 29, 2018 - Python
PyTorch-Code for the Paper "SmaAt-UNet: Precipitation Nowcasting using a Small, Attentive UNet-Architecture"
- Updated
Mar 13, 2025 - Python
This repository implements pytorch version of the modifed 3D U-Net from Fabian Isensee et al. participating in BraTS2017
- Updated
Sep 7, 2018 - Python
"pip install unet": PyTorch Implementation of 1D, 2D and 3D U-Net architecture.
- Updated
Dec 13, 2024 - Python
A PyTorch 1.0 Implementation of Unet with EfficientNet as encoder
- Updated
Mar 13, 2021 - Python
Open solution to the Data Science Bowl 2018
- Updated
Jun 22, 2022 - Python
Segmentation deep learning ALgorithm based on MONai toolbox: single and multi-label segmentation software developed by QIMP team-Vienna.
- Updated
Nov 15, 2021 - Python
Automatic information extraction from identity card with ocr
- Updated
May 16, 2024 - Python
This project is about detecting defects on steel surface using Unet. The dataset used for this project is the NEU-DET database.
- Updated
May 22, 2021 - Jupyter Notebook
Liver Lesion Segmentation with 2D Unets
- Updated
Jul 4, 2019 - Python
This repository contains the code for semantic segmentation of the retina blood vessel on the DRIVE dataset using the PyTorch framework.
- Updated
Jul 12, 2021 - Python
Can you detect and classify defects in steel? Segmentation in Pytorch
- Updated
Sep 18, 2020 - Python
Improve this page
Add a description, image, and links to theunet-pytorch topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with theunet-pytorch topic, visit your repo's landing page and select "manage topics."