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-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.

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.

a-unet

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

chuchichaestli

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

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

unetpy

UnetSocket: A high-level socket interface for UnetStack in Python

arlpy

ARL Python Tools

metnet3-pytorch

MetNet 3 - Pytorch

keras-unet

Helper package with multiple U-Net implementations in Keras as well as useful utility tools helpful when working with image segmentation tasks

uformer-pytorch

Implementation of Uformer, Attention-based Unet, in Pytorch

seg-torch

Semantic Segmentation with Pytorch

covidxai

Implementation of explainability algorithms (layer-wise relevance propagation, local interpretable model-agnostic explanations, gradient-weighted class activation mapping) on computer vision architectures to identify and explain regions of COVID 19 pneumonia in chest X-ray and CT scans.

denoisers

A package for training audio denoisers

efficientunet-pytorch

A PyTorch 1.0 Implementation of Unet with EfficientNet as encoder

bc-count

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

flatiron

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

livermask

A package for automatic segmentation of liver from CT data

torch-segment

A library for deep learning image segmentation using PyTorch

segmentation-models-zwy

Segmentation models with pretrained backbones. Keras and TensorFlow Keras.

efficientunet

Keras Implementation of Unet with EfficientNet as encoder.