U Net Python Packages

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

x-unet

Implementation of a U-net complete with efficient attention as well as the latest research findings

rankseg

Boost segmentation model mIoU/Dice instantly WITHOUT retraining. A plug-and-play, training-free optimization module. Published in NeurIPS & JMLR. Compatible with SAM, DeepLab, SegFormer, and more. 🧩

ct-segmentation-toolkit

Supervised (U-Net), unsupervised (Otsu, k-means, GMM), and label-free spectral-spatial (SOM) segmentation of scientific image stacks, with self-supervised Noise2Inverse denoising.

unet-stylegan2

A Pytorch implementation of Stylegan2 with UNet Discriminator

trixi

Manage your machine learning experiments with trixi - modular, reproducible, high fashion. An experiment infrastructure optimized for PyTorch, but flexible enough to work for your framework and your tastes.

trixi-slim

Manage your machine learning experiments with trixi - modular, reproducible, high fashion. An experiment infrastructure optimized for PyTorch, but flexible enough to work for your framework and your tastes.

modular-diffusion

Python library for designing and training your own Diffusion Models with PyTorch

train-unet

Low-code medical image semantic Segmentation using U-net architecture

metrics-petri

Python package repository for quantifying Magnaporthe oryzae colony growth, morphology, cracking, and texture from time-series Petri-dish images.

fluospotter

Fluospotter is an end-to-end pipeline designed for nuclei segmentation and puncta detection in fluorescence microscopy images.