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

metrics-petri

Python package for analysing Magnaporthe oryzae growth on petri dishes

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

x-unet

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

unet-stylegan2

StyleGan2 with UNet Discriminator, in Pytorch

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.

train-unet

To use U-net architecture and train models for medical imaging at ease.

trixi

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

promethium-seismic

Promethium is a state-of-the-art, AI-driven framework for seismic signal reconstruction, denoising, and geophysical data enhancement, integrating cutting-edge deep learning architectures with production-grade data engineering.

modular-diffusion

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

fluospotter

Python library designed for nuclei segmentation and puncta detection in fluorescence microscopy images.