Shypd.aiThe largest open index of AI toolsbatchgenerators is an open-source data augmentation framework for 2D and 3D image classification and segmentation. It supports a variety of spatial, color, and noise augmentations compatible with both 2D and 3D input data.
Trending
About
batchgenerators is a Python package designed for data augmentation, specifically tailored for 2D and 3D image classification and segmentation tasks. Developed jointly by the German Cancer Research Center (DKFZ) and the Helmholtz Imaging Platform, it offers a comprehensive suite of augmentations including mirroring, channel translation, elastic deformations, rotations, scaling, resampling, and multi-channel misalignments for spatial data. Color augmentations cover brightness, contrast, and gamma, while noise augmentations include Gaussian and Rician noise. The framework also provides cropping options like random and center crop, along with padding. A key differentiator is its compatibility with both 2D and 3D input data, addressing a common gap in other frameworks. It also features anatomy-informed and misalignment data augmentations for specialized applications. The package is designed for flexibility, using a simple Python dictionary structure for data handling, and supports multi-threaded augmentation for performance.
Capabilities
Pricing & Plans
Open Source
Free
FAQs
Trending
