Search Packages

segmentation-models-pytorch

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

unet

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

d3m-segmentation-models-pytorch

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

segmentation-models-pytorch-deepflash2

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

bio-image-unet

Implementation of U-Net, U-Net++, and 3D U-Net in PyTorch for analysis of bio-images

efficientunet-pytorch

A PyTorch 1.0 Implementation of Unet with EfficientNet as encoder

cospgd

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

torch-segment

A library for deep learning image segmentation using PyTorch

pytorch3dunet

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

pytorch3dunet-spotmax

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

pytorch3dunet-smax

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

unet-pytorch

PyTorch implementation of a convolutional neural network (U-Net)

segmentation-sust

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