GitHub topics: unet-pytorch

Repositories

wolny/pytorch-3dunet

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

Language: Jupyter Notebook - Size: 291 MB - Last synced at: about 1 hour ago - Pushed at: about 2 months ago - Stars: 2,166 - Forks: 529

qubvel-org/segmentation_models.pytorch

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

Language: Python - Size: 14.1 MB - Last synced at: 1 day ago - Pushed at: 3 days ago - Stars: 10,345 - Forks: 1,736

pollockjj/ComfyUI-MultiGPU Fork of neuratech-ai/ComfyUI-MultiGPU

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.

Language: Python - Size: 14.1 MB - Last synced at: 3 days ago - Pushed at: 3 days ago - Stars: 212 - Forks: 9

danihae/bio-image-unet

U-Net implementation in PyTorch for segmentation of bio-images

Language: Python - Size: 251 KB - Last synced at: 5 days ago - Pushed at: 5 days ago - Stars: 4 - Forks: 0

yakhyo/crack-segmentation

🛣️🔍 | Road crack segmentation using UNet in PyTorch > Implementation of different loss functions (i.e Focal, Dice, Dice + CE)

Language: Jupyter Notebook - Size: 107 MB - Last synced at: 6 days ago - Pushed at: 6 days ago - Stars: 27 - Forks: 4

giansimone/unet-pytorch

PyTorch implementation of a convolutional neural network (U-Net) for semantic segmentation of biomedical images.

Language: Python - Size: 85.9 KB - Last synced at: 8 days ago - Pushed at: 11 days ago - Stars: 0 - Forks: 0

HansBambel/SmaAt-UNet

PyTorch-Code for the Paper "SmaAt-UNet: Precipitation Nowcasting using a Small, Attentive UNet-Architecture"

Language: Python - Size: 914 KB - Last synced at: 8 days ago - Pushed at: about 1 month ago - Stars: 252 - Forks: 48

Koldim2001/Unet-pytorch-training

Репозиторий для обучения нейросетевых моделей по семантической сегментации + пример использования моделей на практике

Language: Jupyter Notebook - Size: 55.1 MB - Last synced at: 8 days ago - Pushed at: 3 months ago - Stars: 31 - Forks: 1

AnshKGoyal/Image-Enhancer

Hybrid Image Enhancement platform combining FastAPI and Streamlit for customizable multi-stage pipelines using deep learning. Supports flexible deblurring and dehazing workflows, user authentication, and output history. Easily extendable architecture with pre-trained model weights for advanced image restoration.

Language: Jupyter Notebook - Size: 18.1 MB - Last synced at: 16 days ago - Pushed at: 16 days ago - Stars: 0 - Forks: 0

Mqtth3w/pixel-level-segmentation-in-pets-unipr

UNet and DeepLabV3 for instance segmentation. Starting from a rgb image of a pet the network should output a segmentation mask of the pet. Deep Learning and Generative Models' practical project developed in 2025.

Language: Jupyter Notebook - Size: 3.8 MB - Last synced at: 16 days ago - Pushed at: 16 days ago - Stars: 0 - Forks: 0

CedricCaruzzo/pannuke-segmentation

An open-source UNet-based pipeline for nuclei segmentation in histopathology images using the PanNuke dataset. It features an interactive web app for easy data visualization and handling, making AI tools accessible even for non-experts. This project provides a foundation for training and exploring histopathology data.

Language: Python - Size: 107 KB - Last synced at: 16 days ago - Pushed at: 3 months ago - Stars: 6 - Forks: 4

Mostafa-wael/U-Net-in-PyTorch

This is an implementation of the U-Net model from the paper, U-Net: Convolutional Networks for Biomedical Image Segmentation.

Language: Jupyter Notebook - Size: 433 KB - Last synced at: 6 days ago - Pushed at: almost 2 years ago - Stars: 23 - Forks: 8

sdsubhajitdas/Brain-Tumor-Segmentation

Brain Tumor Segmentation done using U-Net Architecture.

Language: Jupyter Notebook - Size: 37.1 MB - Last synced at: 14 days ago - Pushed at: almost 2 years ago - Stars: 279 - Forks: 63

SvenPfiffner/SkinMachine

SkinMachine, a simple UNet diffusion model to generate novel Minecraft skins at random.

Language: Jupyter Notebook - Size: 2.16 MB - Last synced at: 22 days ago - Pushed at: 22 days ago - Stars: 0 - Forks: 0

wanghao6736/UNet-with-Pytorch

基于 U-Net 的图像分割模型

Language: Python - Size: 142 KB - Last synced at: 27 days ago - Pushed at: 27 days ago - Stars: 0 - Forks: 0

tugstugi/pytorch-saltnet

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

Language: Python - Size: 732 KB - Last synced at: 18 days ago - Pushed at: over 6 years ago - Stars: 279 - Forks: 52

nikhilroxtomar/Retina-Blood-Vessel-Segmentation-in-PyTorch

This repository contains the code for semantic segmentation of the retina blood vessel on the DRIVE dataset using the PyTorch framework.

Language: Python - Size: 8.59 MB - Last synced at: 7 days ago - Pushed at: almost 4 years ago - Stars: 70 - Forks: 28

matin-ghorbani/Salt-Segmentation-UNet

Implement a salt segmentation using PyTorch for TGS Salt Identification Challenge

Language: Python - Size: 338 KB - Last synced at: 17 days ago - Pushed at: 11 months ago - Stars: 12 - Forks: 0

Charmve/Semantic-Segmentation-PyTorch

PyTorch implementation for Semantic Segmentation, include FCN, U-Net, SegNet, GCN, PSPNet, Deeplabv3, Deeplabv3+, Mask R-CNN, DUC, GoogleNet, and more dataset

Language: Python - Size: 2.72 MB - Last synced at: 8 days ago - Pushed at: 11 months ago - Stars: 55 - Forks: 10

nata-vito/skin-cancer-segmentation

The aim of this project is to train segmentation models with png mask. This work uses U-Net model specialized in biomedical image segmentation in order to segment skin cancers.

Language: Jupyter Notebook - Size: 7.69 MB - Last synced at: 22 days ago - Pushed at: about 1 month ago - Stars: 4 - Forks: 0

negarhonarvar/DAGMRNet

A dynamic attentive graph model for cardiac MRI image reconstrunction of CMRxRecon dataset with PromptUnet for sensitivity map estimation.

Language: Python - Size: 71 MB - Last synced at: about 1 month ago - Pushed at: about 1 month ago - Stars: 4 - Forks: 0

ababier/deep-dicom

Preprocess and standardize DICOM data for deep learning applications in radiotherapy.

Language: Python - Size: 34.2 KB - Last synced at: about 1 month ago - Pushed at: about 1 month ago - Stars: 0 - Forks: 0

neptune-ai/open-solution-mapping-challenge

Open solution to the Mapping Challenge :earth_americas:

Language: Python - Size: 704 KB - Last synced at: 6 days ago - Pushed at: about 4 years ago - Stars: 386 - Forks: 97

laoluadewoye/Various-Dim-Unet-ResNet

Library for easy usage of general purpose and diffusion UNETs for any dimension of data.

Language: Jupyter Notebook - Size: 842 KB - Last synced at: about 2 months ago - Pushed at: about 2 months ago - Stars: 0 - Forks: 0

zhoudaxia233/EfficientUnet-PyTorch

A PyTorch 1.0 Implementation of Unet with EfficientNet as encoder

Language: Python - Size: 25.4 KB - Last synced at: 11 days ago - Pushed at: about 4 years ago - Stars: 158 - Forks: 33

ksasso1028/indi-debanding

Image Debanding using Inversion by Direct Iteration

Language: Python - Size: 65.4 MB - Last synced at: 16 days ago - Pushed at: 11 months ago - Stars: 4 - Forks: 1

hiyouga/Image-Segmentation-PyTorch

U-Net for image segmentation, PyTorch implementation.

Language: Python - Size: 96 MB - Last synced at: 18 days ago - Pushed at: over 4 years ago - Stars: 17 - Forks: 4

tangezerman/U-net-Tissue-Mask-Generation

A VGG16 backed U-Net model that generates binary masks out of high resolution whole slide images for histopathologists.

Language: Python - Size: 111 KB - Last synced at: 3 months ago - Pushed at: 3 months ago - Stars: 0 - Forks: 0

caiyunapp/leibniz

Leibniz is a python package which provide facilities to express learnable partial differential equations with PyTorch

Language: Python - Size: 1.46 MB - Last synced at: 5 days ago - Pushed at: over 3 years ago - Stars: 15 - Forks: 7

davidiommi/Pytorch--3D-Medical-Images-Segmentation--SALMON 📦

Segmentation deep learning ALgorithm based on MONai toolbox: single and multi-label segmentation software developed by QIMP team-Vienna.

Language: Python - Size: 15.4 MB - Last synced at: 8 days ago - Pushed at: over 3 years ago - Stars: 121 - Forks: 26

MilleXi/Retinal-vessel-segmentation

:fire:一个基于 UNet 的视网膜血管分割项目，使用 PyTorch 实现并基于 DRIVE 数据集进行训练和测试。项目包括完整的数据处理、模型训练和测试流程，最终生成视网膜图像的分割结果。

Language: Python - Size: 2.93 KB - Last synced at: 4 months ago - Pushed at: 4 months ago - Stars: 0 - Forks: 0

GohVh/resnet34-unet

Segmentation model using UNET architecture with ResNet34 as encoder background, designed with PyTorch.

Language: Python - Size: 1.74 MB - Last synced at: 4 months ago - Pushed at: over 2 years ago - Stars: 8 - Forks: 2

bhimrazy/unet

PyTorchUNet : A PyTorch Implementation of UNet Architecture for Semantic Segmentation of Images from scratch.

Language: Python - Size: 16 MB - Last synced at: 1 day ago - Pushed at: over 1 year ago - Stars: 4 - Forks: 1

prasannaghimiree/Breast-Cancer-Histopathology-Image-Segmentation-using-U-Net

Breast cancer histopathology image segmentation using U-Net. This repository implements U-Net for accurate segmentation of cancerous regions. It includes data augmentation, mixed precision training, checkpointing, and evaluation metrics like Dice score to improve model performance.

Language: Jupyter Notebook - Size: 2.6 MB - Last synced at: 2 months ago - Pushed at: 5 months ago - Stars: 0 - Forks: 0

abarton51/Hyperspectral-Deep-Deconvolution

Project on using deep models for deconvolution of hyperspectral images with chromatic aberration.

Language: TeX - Size: 142 MB - Last synced at: 6 months ago - Pushed at: 6 months ago - Stars: 1 - Forks: 0

shashankskagnihotri/cospgd

The official repository for CosPGD: a unified white-box adversarial attack for pixel-wise prediction tasks.

Language: Python - Size: 937 KB - Last synced at: 10 days ago - Pushed at: 8 months ago - Stars: 11 - Forks: 1

zhoudaxia233/PyTorch-Unet

PyTorch1.0 Implementation of Unet

Language: Python - Size: 103 KB - Last synced at: 19 days ago - Pushed at: almost 6 years ago - Stars: 16 - Forks: 5

Akaqox/unet-segmentation-with-docker

U-Net segmentation algorithm with options of pretrained resnet34 and resnet50 encoders. All of the project dockerized with gpu suppport on anaconda environment with multiple loss support..

Language: Python - Size: 19.5 KB - Last synced at: 16 days ago - Pushed at: 7 months ago - Stars: 1 - Forks: 0

rishikksh20/ResUnet

Pytorch implementation of ResUnet and ResUnet ++

Language: Python - Size: 33.2 KB - Last synced at: 8 months ago - Pushed at: 8 months ago - Stars: 448 - Forks: 66

deepmancer/unet-semantic-segmentation

PyTorch UNet model for semantic segmentation of urban scenes using the Cityscapes dataset

Language: Jupyter Notebook - Size: 2.51 MB - Last synced at: 6 months ago - Pushed at: 8 months ago - Stars: 3 - Forks: 0

callmetandat/Parallelizing-U-Net

Language: Jupyter Notebook - Size: 17.9 MB - Last synced at: 8 months ago - Pushed at: 8 months ago - Stars: 0 - Forks: 0

Youngsiii/UNet_KANConv

Bottleneck KANConv for Unet

Language: Python - Size: 1.04 MB - Last synced at: 9 months ago - Pushed at: 10 months ago - Stars: 2 - Forks: 0

khanhdq109/Lung-Tumor-Segmentation

Segmentation of a small target (cancer) in a large image

Language: Jupyter Notebook - Size: 3.17 MB - Last synced at: 3 days ago - Pushed at: over 2 years ago - Stars: 3 - Forks: 0

minerva-ml/open-solution-data-science-bowl-2018 📦

Open solution to the Data Science Bowl 2018

Language: Python - Size: 7.03 MB - Last synced at: 10 months ago - Pushed at: almost 3 years ago - Stars: 158 - Forks: 42

enesdoruk/DomAtt-UNet

Domain Adaptive Based Semantic Segmentation

Language: Python - Size: 152 KB - Last synced at: 11 months ago - Pushed at: 11 months ago - Stars: 0 - Forks: 0

huichen5796/2022-studienarbeit-hui-chen

a tool for detecting tables in image and analysing complex header

Language: Python - Size: 874 MB - Last synced at: about 2 months ago - Pushed at: 11 months ago - Stars: 3 - Forks: 0

aterelak/ANN-unet

The repo of the ANN's class final project in NCU (Toruń, Poland). It is an implementation of the paper "U-Net: Convolutional Networks for Biomedical Image Segmentation".

Language: Jupyter Notebook - Size: 343 MB - Last synced at: 11 months ago - Pushed at: 11 months ago - Stars: 0 - Forks: 0

yakhyo/unet-pytorch

🚗 | UNet implementation using PyTorch | CARVANA Dataset | Car Segmentation

Language: Jupyter Notebook - Size: 319 MB - Last synced at: 11 months ago - Pushed at: 11 months ago - Stars: 4 - Forks: 0

chathumal93/Pytorch-UNet-Flood-Segmentation

This repository contains a PyTorch implementation of a U-Net model for segmenting water areas (flood and permananet water) in Sentinel-1 satellite images.

Language: Jupyter Notebook - Size: 53.8 MB - Last synced at: 11 months ago - Pushed at: 11 months ago - Stars: 13 - Forks: 4

op1009/flood_area_segmentation

Segment flood area from aerial RGB images

Language: Jupyter Notebook - Size: 774 KB - Last synced at: 11 months ago - Pushed at: 11 months ago - Stars: 0 - Forks: 0

musimab/Tc_ID_Card_OCR

Automatic information extraction from identity card with ocr

Language: Python - Size: 164 MB - Last synced at: 11 months ago - Pushed at: 11 months ago - Stars: 74 - Forks: 23

avulaankith/Highway-lane-segmentation

Highway lane segmentation Dataset

Language: Jupyter Notebook - Size: 1.04 MB - Last synced at: 11 months ago - Pushed at: 11 months ago - Stars: 0 - Forks: 0

Highdrien/segmentation_MRI

semantic segmentation of cardiac MRI to detect different parts of the heart

Language: Python - Size: 35.2 KB - Last synced at: 11 months ago - Pushed at: about 2 years ago - Stars: 0 - Forks: 0

ASEM000/UNet-Medical-Segmentation-Decathlon

UNet implementation to segment left ventricle in pytorch

Language: Jupyter Notebook - Size: 230 MB - Last synced at: 28 days ago - Pushed at: over 3 years ago - Stars: 4 - Forks: 0

AImageLab-zip/alveolar_canal

This repository contains the material from the paper "Improving Segmentation of the Inferior Alveolar Nerve through Deep Label Propagation"

Language: Python - Size: 973 KB - Last synced at: 12 months ago - Pushed at: 12 months ago - Stars: 34 - Forks: 4

fepegar/unet

"pip install unet": PyTorch Implementation of 1D, 2D and 3D U-Net architecture.

Language: Python - Size: 89.8 KB - Last synced at: 12 months ago - Pushed at: over 1 year ago - Stars: 134 - Forks: 22

DebeshJha/TransRUPNet

TransRUPNet for Improved Out-of-Distribution Generalization in Polyp Segmentation (IEEE EMBC)

Language: Python - Size: 827 KB - Last synced at: about 1 year ago - Pushed at: about 1 year ago - Stars: 4 - Forks: 0

ry-lu/pytorch-face-reaging-network

PyTorch Lightning implementation of Disney's face re-aging network (FRAN) paper.

Language: Python - Size: 39.1 MB - Last synced at: about 1 year ago - Pushed at: about 1 year ago - Stars: 0 - Forks: 0

05kashyap/Lung-Tumor-Segmentation-DS

Lung tumor segmentation with the UNet model.

Language: Jupyter Notebook - Size: 3.95 MB - Last synced at: about 1 year ago - Pushed at: about 1 year ago - Stars: 1 - Forks: 0

mjDelta/CV-in-ADAS-pytorch

This repo includes Unet, Spatial CNN (S-CNN) and VPNet for lane segmentation, and YOLO, Faster-RCNN, Stereo-RCNN for vehicle detection.

Language: Jupyter Notebook - Size: 96 MB - Last synced at: about 1 year ago - Pushed at: almost 6 years ago - Stars: 27 - Forks: 4

KrisnaPinasthika/Deciphering-Pixel-Insights

Official implementation of Deciphering pixel insights: A deep dive into deep learning strategies for enhanced indoor depth estimation

Language: Jupyter Notebook - Size: 21.1 MB - Last synced at: about 1 year ago - Pushed at: about 1 year ago - Stars: 0 - Forks: 0

ArghyaChatterjee/unet-nuclear-thermal-ir-image-segmentation

Qualitative & quantitative performance evaluation of traditional image processing and CNN based automatic methods on Pool Boiling IR images of MIT’s Nuclear Reactor Lab by measuring important boiling heat transfer parameters statistically. Work performed in 2019 as part of BSc. Thesis.

Language: Jupyter Notebook - Size: 2.84 MB - Last synced at: about 1 year ago - Pushed at: about 1 year ago - Stars: 2 - Forks: 0

FrexG/ELU-Net-pytorch

Pytorch implementation of ELU-Net: An Efficient and Lightweight U-Net for Medical Image Segmentation

Language: Jupyter Notebook - Size: 376 KB - Last synced at: 12 months ago - Pushed at: over 1 year ago - Stars: 6 - Forks: 2

Nobodyworld/De-Noisy-Image-Project-Xtra-Small

The extra small and even more CPU friendly version of the "De-Noisy Image Project." This repository contains scripts and directories for a public domain image editing neural network, 'De-Noisy Image Project', using a U-Net architecture.

Language: Python - Size: 1.55 GB - Last synced at: about 1 year ago - Pushed at: about 1 year ago - Stars: 0 - Forks: 0

Nobodyworld/De-Noisy-Image-Project-Small

The smaller and more CPU friendly version of the De-Noisy Image Project. This repository contains scripts and directories for a public domain image editing neural network, 'De-Noisy Image Project', using a U-Net architecture.

Language: Python - Size: 2.15 GB - Last synced at: about 1 year ago - Pushed at: about 1 year ago - Stars: 0 - Forks: 0