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.

chuchichaestli

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

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

denoisers

Simple PyTorch Denoisers for Waveform Audio

a-unet

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

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.

unet

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

arlpy

ARL Python Tools

unetpy

UnetSocket interfaces for various languages

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

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

metnet3-pytorch

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

seg-torch

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

sentinel-crop-pipeline

Open-source, config-driven pipeline for discovering, preprocessing, and patching Sentinel-2 L2A imagery into U-Net-ready training datasets for crop classification — hybrid rule-based + AI scene selection with ground-truth label integration.

mouse-pupil-analysis

A mouse pupil size analysis pipeline in a single line of command. Works well in low-contrast and low-resolution frames.

pytorch3dunet

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

mridc

Data Consistency Toolbox for Magnetic Resonance Imaging

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.

uformer-pytorch

Implementation of Uformer, Attention-based Unet, in Pytorch

livermask

💥 Command line tool for automatic liver parenchyma and liver vessel segmentation in CT using a pretrained deep learning model

efficientunet-pytorch

A PyTorch 1.0 Implementation of Unet with EfficientNet as encoder

segmentation-models-pytorch-deepflash2

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