Registry / ai-ml / batchgenerators

batchgenerators

library0.25.3pypypi✓ verified 80d ago

Data augmentation toolkit for medical imaging and deep learning. Current version 0.25.1. Released on PyPI with irregular cadence; maintained by MIC-DKFZ.

ai-ml data

pip install batchgenerators

github ↗package ↗

INSTALL

IMPORT

SIG · BATCHGENERATORS

batchgenerators

ai-mlpythonv0.25.3

Install

17.2s avg

Import

—

Disk

435MB

Pass rate

5/ 10

Env Coverage5 / 10

glibc

3.9–3.13

musl

3.9–3.13

Install & Compatibility

Where this runs

tested against v0.25.3 · pip install

no network on importno background threads

Install × environment matrix

Each cell = how many times install + import succeeded across repeated harness runs. Partial = flaky.
glibc = Debian/Ubuntu slim · musl = Alpine Linux

musl

py 3.10–3.95 runs

build_error

glibc

py 3.10–3.95 runs

installs and imports cleanly · install 17.2s · import 0.000s · 423MB

435MB installed

● package 435MB

Code

Verified usage

Verified import paths — ran on the pinned version, not inferred.

DataLoader

✓ from batchgenerators.dataloading.data_loader import DataLoader

✗ from batchgenerators.dataloading import DataLoader

Basic usage: create a DataLoader from a list of data, apply augmentations via Compose.

from batchgenerators.dataloading import DataLoader from batchgenerators.transforms import Compose, SpatialTransform # Simple data loader for a list of images train_loader = DataLoader(my_data_list, batch_size=4, num_threads=2) transforms = Compose([SpatialTransform(patch_size=(128,128,128))]) for batch in train_loader: data = batch['data'] seg = batch['seg'] data, seg = transforms(data, seg) break

Debug

Known issues

breakingIn version 0.25, DataLoader no longer accepts the 'dataset' argument; use 'data' instead.

fix

Replace DataLoader(dataset=...) with DataLoader(data=...)

affects: >=0.25

deprecatedbatchgenerators.transforms.abstract_transforms is deprecated; use batchgenerators.transforms directly.

fix

Import from batchgenerators.transforms instead.

affects: >=0.23

gotchaMemory usage can spike if num_threads is set too high in DataLoader, especially with large images.

fix

Keep num_threads <= number of CPU cores; use multiprocessing context if needed.

affects: all

Upgrade

Version history

0.25.3latest on PyPI · released Jun 16, 2026

Audit

Dependencies

No dependency data recorded yet.

Agent activity

23 hits · last 30 days

node

OpenAI (training)

Resources

githubhttps://github.com/MIC-DKFZ/batchgenerators ↗

packagehttps://pypi.org/project/batchgenerators/ ↗