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

3D U-Net model for volumetric semantic segmentation written in pytorch

Pytorch implementation of ResUnet and ResUnet ++

Open solution to the Mapping Challenge 🌎

Official repo for Medical Image Segmentation Review: The Success of U-Net

A Probabilistic U-Net for segmentation of ambiguous images implemented in PyTorch

Kaggle | 9th place single model solution for TGS Salt Identification Challenge

Brain Tumor Segmentation done using U-Net Architecture.

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.

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

"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.

This project is about detecting defects on steel surface using Unet. The dataset used for this project is the NEU-DET database.

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.

Can you detect and classify defects in steel? Segmentation in Pytorch