Unet Python Packages

segmentation-models-pytorch

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

segmentation-models

Segmentation models with pretrained backbones. Keras and TensorFlow Keras.

unet

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

keras-unet

Helper package with multiple U-Net implementations in Keras as well as useful utility tools helpful when working with image semantic segmentation tasks. This library and underlying tools come from multiple projects I performed working on semantic segmentation tasks

keras-unet-collection

The Tensorflow, Keras implementation of U-net, V-net, U-net++, UNET 3+, Attention U-net, R2U-net, ResUnet-a, U^2-Net, TransUNET, and Swin-UNET with optional ImageNet-trained backbones.

a-unet

A toolbox that provides hackable building blocks for generic 1D/2D/3D UNets, in PyTorch.

arlpy

ARL Python Tools

chuchichaestli

Where you find all the state-of-the-art cooking utensils (salt, pepper, gradient descent... the usual).

metnet3-pytorch

Implementation of MetNet-3, SOTA neural weather model out of Google Deepmind, in Pytorch

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.

uformer-pytorch

Implementation of Uformer, Attention-based Unet, in Pytorch

segmentation-models-zwy

Segmentation models with pretrained backbones. Keras and TensorFlow Keras.

swiss-army-keras

A library to help with the development of AI models with Keras, with a focus on edge / IoT applications. Based originally on https://github.com/yingkaisha/keras-unet-collection

efficientunet-pytorch

A PyTorch 1.0 Implementation of Unet with EfficientNet as encoder

unetpy

UnetSocket interfaces for various languages

seg-torch

Pytorch implementation of FCN, UNet, PSPNet, and various encoder models.

pytorch3dunet

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

denoisers

Simple PyTorch Denoisers for Waveform Audio

bc-count

Count red, white blood cells to detect various diseases such as blood cancer (leukemia), lower red blood cells (anemia)...

torch-segment

A deep learning image segmentation library and API on top of PyTorch.

flatiron

Standardized computer vision pipeline framework with a chainable API and procedurally defined model architecture.

unet-pytorch

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

efficientunet

Keras Implementation of Unet with EfficientNet as encoder