unet

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

MIT License

Downloads
1.8K
Stars
134
Committers
6
View Code on GitHub View on X
Ecosystems: Python

Bot releases are visible (Hide)

unet - PyTorch implementation of 2D and 3D U-Net Latest Release

Published by fepegar over 4 years ago

unet - PyTorch implementation of U-Net

Published by fepegar over 4 years ago

unet - First published version of PyTorch U-Net

Published by fepegar almost 5 years ago

Package Rankings
Top 7.48% on Pypi.org
Related Projects
3DUnetCNN

Pytorch 3D U-Net Convolution Neural Network (CNN) designed for medical image segmentation

01 Feb 2017 1,888
cytounet

Deep Learning Based Cell Segmentation

16 Jun 2020 6
vesseg

Brain vessel segmentation using 3D convolutional neural networks

26 Nov 2018 33
uformer-pytorch

Implementation of Uformer, Attention-based Unet, in Pytorch

17 Jun 2021 93
Unet-ants

UNet deep learning model for medical image segmentation

31 Aug 2017 45
U-2-Net

The code for our newly accepted paper in Pattern Recognition 2020: "U^2-Net: Going Deeper with Ne...

04 May 2020 8,393
MRISkullStripping

Developing a UNet3D model for accurate MRI skull stripping using the Calgary Campinas 359 dataset...

18 Aug 2024 88
unet-stylegan2

A Pytorch implementation of Stylegan2 with UNet Discriminator

25 Jul 2020 199
image-segmentation-keras

Implementation of Segnet, FCN, UNet , PSPNet and other models in Keras.

20 Mar 2017 2,902
tensorrt-examples

TensorRT Examples (TensorRT, Jetson Nano, Python, C++)

28 Jun 2021 71
highresnet

PyTorch implementation of HighRes3DNet

05 Jul 2019 49
Unet-Segmentation-Pytorch-Nest-of-Unets

Implementation of different kinds of Unet Models for Image Segmentation - Unet , RCNN-Unet, Atten...

14 May 2019 1,862
TransUNet

This repository includes the official project of TransUNet, presented in our paper: TransUNet: Tr...

08 Feb 2021 2,346
Image_Segmentation

Pytorch implementation of U-Net, R2U-Net, Attention U-Net, and Attention R2U-Net.

18 Jun 2018 2,690
UNetPlusPlus

[IEEE TMI] Official Implementation for UNet++

16 Apr 2018 2,273