2D single generator

class biapy.data.generators.single_data_2D_generator.Single2DImageDataGenerator(**kwars)[source]

Bases: SingleBaseDataGenerator

Custom 2D data generator based on imgaug and our own augmentors.py transformations. This generator will yield an image and its corresponding class.

Based on microDL and Shervine’s blog.

ensure_shape(img)[source]
save_aug_samples(img, orig_image, i, pos, out_dir, draw_grid)[source]