unet-pytorch

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

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

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

Brain Tumor Segmentation done using U-Net Architecture.

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

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