unet-pytorch

There are 2 repositories under unet-pytorch topic.

qubvel-org / segmentation_models.pytorch
Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.
computer-vision deeplab-v3-plus deeplabv3 dpt fpn image-processing image-segmentation imagenet models pretrained-weights pspnet pytorch segformer segmentation segmentation-models semantic-segmentation transformers unet unet-pytorch unetplusplus
Language:Python 10885
wolny / pytorch-3dunet
3D U-Net model for volumetric semantic segmentation written in pytorch
semantic-segmentation unet unet-pytorch 3d-unet groupnorm pytorch-3dunet pytorch dice-coefficient volumetric-data 3d-segmentation residual-unet
Language:Jupyter Notebook 2251
rishikksh20 / ResUnet
Pytorch implementation of ResUnet and ResUnet ++
deep-residual-network image-processing image-segmentation image-to-image-translation medical-image-processing resunet satellite-imagery segmentation unet unet-pytorch
Language:Python 518
pollockjj / ComfyUI-MultiGPU
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
comfyui comfyui-nodes comfyui-workflow pytorch stable-diffusion ggml gguf-models unet-pytorch wanvideowrapper
Language:Python 479
neptune-ai / open-solution-mapping-challenge
Open solution to the Mapping Challenge :earth_americas:
data-science machine-learning deep-learning kaggle python satellite-imagery data-science-learning lightgbm unet unet-image-segmentation unet-pytorch neptune machine-learning-lab mapping-challenge crowdai competition pipeline pipeline-framework
Language:Python 387
NITR098 / Awesome-U-Net
Official repo for Medical Image Segmentation Review: The Success of U-Net
medical-image-processing medical-image-segmentation segmentation-models transformer u-net unet-image-segmentation unet-pytorch
Language:Jupyter Notebook 341
stefanknegt / Probabilistic-Unet-Pytorch
A Probabilistic U-Net for segmentation of ambiguous images implemented in PyTorch
unet-image-segmentation unet-pytorch
Language:Python 288
sdsubhajitdas / Brain-Tumor-Segmentation
Brain Tumor Segmentation done using U-Net Architecture.
deep-learning image-segmentation pytorch unet unet-image-segmentation unet-pytorch tumor-segmentation brain-tumor-segmentation
Language:Jupyter Notebook 286
tugstugi / pytorch-saltnet
Kaggle | 9th place single model solution for TGS Salt Identification Challenge
deep-learning cnn pytorch segmentation convolutional-neural-networks unet unet-image-segmentation unet-pytorch seismic-imaging kaggle-competition
Language:Python 281
HansBambel / SmaAt-UNet
PyTorch-Code for the Paper "SmaAt-UNet: Precipitation Nowcasting using a Small, Attentive UNet-Architecture"
convolutional-neural-networks precipitation-nowcasting unet unet-pytorch weather-forecast
Language:Python 272
pykao / Modified-3D-UNet-Pytorch
This repository implements pytorch version of the modifed 3D U-Net from Fabian Isensee et al. participating in BraTS2017
unet-pytorch 3d-unet unet unet-image-segmentation unet-3d
Language:Python 220
fepegar / unet
"pip install unet": PyTorch Implementation of 1D, 2D and 3D U-Net architecture.
unet-image-segmentation unet unet-pytorch deep-learning convolutional-neural-networks medical-imaging
Language:Python 174
zhoudaxia233 / EfficientUnet-PyTorch
A PyTorch 1.0 Implementation of Unet with EfficientNet as encoder
unet unet-pytorch image-segmentation efficientnet efficientunet
Language:Python 163
minerva-ml / open-solution-data-science-bowl-2018
Open solution to the Data Science Bowl 2018
data-science machine-learning deep-learning kaggle data-science-bowl python image-processing segmentation python3 data-science-bowl-2018 open-source medical-imaging medical-image-processing neptune unet unet-image-segmentation unet-pytorch
Language:Python 156
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.
segmentation monai organs pytorch unet-pytorch unet unet-3d deep-learning monai-toolbox nnunet transformers unetr mri
Language:Python 121
NEU-DET-Steel-Surface-Defect-Detection
siddhartamukherjee / NEU-DET-Steel-Surface-Defect-Detection
This project is about detecting defects on steel surface using Unet. The dataset used for this project is the NEU-DET database.
neu-det defect-detection defect-prediction segmentation-models unet-image-segmentation unet-pytorch resnet resnet-18 pytorch deep-learning deep-neural-networks
Language:Jupyter Notebook 119
musimab / Tc_ID_Card_OCR
Automatic information extraction from identity card with ocr
ocr craft unet-pytorch easyocr id-card-recognition face-recognition computer-vision
Language:Python 106
Confusezius / unet-lits-2d-pipeline
Liver Lesion Segmentation with 2D Unets
deep-learning medical-image-segmentation unet-pytorch lesion-segmentation liver-segmentation unet semantic-segmentation
Language:Python 98
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.
unet unet-pytorch image-segmentation unet-segmentation pytorch drive-dataset retinal-vessel-segmentation
Language:Python 70
Diyago / Severstal-Steel-Defect-Detection
Can you detect and classify defects in steel? Segmentation in Pytorch
deep-learning unet-pytorch unet pytorch python
Language:Python 63
anishreddy3 / Crack-Semantic-Segmentation
Real time crack segmentation using PyTorch, OpenCV and ONNX runtime
deep-learning pytorch opencv onnx unet-pytorch cnn image-processing neural-networks computer-vision crack-detection
Language:Python 61
France1 / unet-multiclass-pytorch
Multiclass semantic segmentation using U-Net architecture combined with strong image augmentation
unet-pytorch unet-image-segmentation biological-image-processing deep-learning pytorch data-augmentation computer-vision multiclass-image-classification
Language:Jupyter Notebook 59
HowieMa / lstm_multi_modal_UNet
[ICIVC 2019] "LSTM multi-modal UNet for Brain Tumor Segmentation"
multi-modal-imaging medical-imaging brats-dataset unet-pytorch lstm-cnn
Language:Python 59
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
semantic-segmentation pytorch pytorch-implementation computer-vision image-processing image-classification segnet unet-pytorch mask-rcnn fcn deep-learning charmve
Language:Python 56
Allenem / MedicalImage
:hospital: 医学图像作业：图像配准论文阅读；眼底血管分割实验
medical-imaging registration segementation unet unet-pytorch unetplusplus
Language:Python 54
josedolz / IVD-Net
Repository containing the source code of the IVD-Net segmentation network that we proposed for the MICCAI 2018 IVD segmentation challenge.
deep-learning medical-imaging medical-image-processing image-segmentation segmentation unet unet-image-segmentation unet-pytorch fcn multi-modal-imaging densenet hyperdensenet miccai pytorch multi-modal-unet intervertebral-disc-localization ivd-net pytorch-cnn
Language:Python 53
Thvnvtos / Lung_Segmentation
Lung Segmentation UNet model on 3D CT scans
unet unet-pytorch lung-segmentation segmentation deep-learning
Language:Python 51
zdaiot / Kaggle-Steel-Defect-Detection
My solution to the Severstal: Steel Defect Detection on Kaggle, which got the 96th place. (Top4%)
kaggle unet-pytorch segmentation
Language:Python 48
DevJake / EEG-diffusion-pytorch
An implementation of Denoising Diffusion Probabilistic Models (DDPM) in PyTorch for EEG-based image reconstruction.
bci brain-computer-interface ddpm deep-learning denoising-diffusion-models eeg electroencephalography generative-model image-reconstruction machine-learning pytorch unet unet-pytorch
Language:Jupyter Notebook 46
lachinov / brats2019
segmentation brats brats2018 brats2019 unet unet-pytorch unet-3d brain-tumor-segmentation
Language:Jupyter Notebook 46
deploy-stable-diffusion-model-on-amazon-sagemaker-endpoint
aws-samples / deploy-stable-diffusion-model-on-amazon-sagemaker-endpoint
Deploy Stable Diffusion Model on Amazon SageMaker Endpont
clip huggingface-transformers-pipeline inference multimodality pytorch sagemaker-deployment sagemaker-pytorch stablediffusion unet-pytorch vae
Language:Jupyter Notebook 39
Koldim2001 / Unet-pytorch-training
Репозиторий для обучения нейросетевых моделей по семантической сегментации + пример использования моделей на практике
camvid-dataset cvat hair-segmentation pytorch-segmentation segmentation segmentation-training semantic-segmentation skin-segmentation streamlit-webapp unet unet-image-segmentation unet-pytorch
Language:Jupyter Notebook 35
alveolar_canal
AImageLab-zip / alveolar_canal
This repository contains the material from the paper "Improving Segmentation of the Inferior Alveolar Nerve through Deep Label Propagation"
segmentation-network unet-pytorch inferior-alveolar-canal inferior-alveolar-nerve
Language:Python 34
Everypixel / arshadowgan-like
ARShadowGAN-like realization. PyTorch.
models python3 pytorch gan neural-networks colab segmentation unet shadow colab-notebook colab-tutorial unet-pytorch shadow-generator arshadowgan
Language:Python 31
gaurav104 / LungSegmentation
This was my research project at IIT Bombay on Lung Segmentation from Chest X-Rays Images
fcn unet unet-pytorch fully-convolutional-networks medical-image-processing chest-xray-images dice-coefficient semantic-segmentation
Language:Python 31
hanyoseob / pytorch-UNET
U-Net: Convolutional Networks for Biomedical Image Segmentation
deep-learning deep-neural-networks convolutional-neural-ne unet unet-pytorch
Language:Python 30

unet-pytorch

qubvel-org / segmentation_models.pytorch

wolny / pytorch-3dunet

rishikksh20 / ResUnet

pollockjj / ComfyUI-MultiGPU

neptune-ai / open-solution-mapping-challenge

NITR098 / Awesome-U-Net

stefanknegt / Probabilistic-Unet-Pytorch

sdsubhajitdas / Brain-Tumor-Segmentation

tugstugi / pytorch-saltnet

HansBambel / SmaAt-UNet

pykao / Modified-3D-UNet-Pytorch

fepegar / unet

zhoudaxia233 / EfficientUnet-PyTorch

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

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

siddhartamukherjee / NEU-DET-Steel-Surface-Defect-Detection

musimab / Tc_ID_Card_OCR

Confusezius / unet-lits-2d-pipeline

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

Diyago / Severstal-Steel-Defect-Detection

anishreddy3 / Crack-Semantic-Segmentation

France1 / unet-multiclass-pytorch

HowieMa / lstm_multi_modal_UNet

Charmve / Semantic-Segmentation-PyTorch

Allenem / MedicalImage

josedolz / IVD-Net

Thvnvtos / Lung_Segmentation

zdaiot / Kaggle-Steel-Defect-Detection

DevJake / EEG-diffusion-pytorch

lachinov / brats2019

aws-samples / deploy-stable-diffusion-model-on-amazon-sagemaker-endpoint

Koldim2001 / Unet-pytorch-training

AImageLab-zip / alveolar_canal

Everypixel / arshadowgan-like

gaurav104 / LungSegmentation

hanyoseob / pytorch-UNET