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

unet-stylegan2

StyleGan2 with UNet Discriminator, in Pytorch

trixi-slim

Manage your machine learning experiments with trixi - modular, reproducible, high fashion

x-unet

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

modular-diffusion

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

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.

train-unet

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

promethium-seismic

Promethium - Advanced Seismic Data Recovery and Reconstruction Framework. State-of-the-art AI-driven framework for seismic signal reconstruction, denoising, and geophysical data enhancement.

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

fluospotter

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