UNet Zoo is a PyTorch library designed to standardize and simplify the usage of various UNet variants in medical image segmentation tasks.
The library provides a unified model API, extensive collection of over 20 UNet variants, config-driven training, and modular design.
UNet Zoo aims to address the challenges of fragmented implementations and APIs of different UNet architectures, ensuring consistency without compromising model-specific features.
The library facilitates production-ready training pipelines, research-friendly experimentation, and aims to accelerate advancements in medical AI tools for applications like tumor detection and surgical planning.