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

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

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

Brain Tumor Segmentation done using U-Net Architecture.

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.

Automatic information extraction from identity card with ocr

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

Liver Lesion Segmentation with 2D Unets

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.

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