MIC-DKFZ / batchgenerators

A framework for data augmentation for 2D and 3D image classification and segmentation
Apache License 2.0
1.09k stars 221 forks source link

Data type for both input images and labels #102

Open zabboud opened 2 years ago

zabboud commented 2 years ago

Does that mean the input has to be a list or must it be an array of arrays based on the example img = np.tile(img[None, None], (self.BATCH_SIZE, 1, 1, 1))?

Also is there anyway of augmenting an image with multiple labels such that the input 'seg' is a list of arrays that must be augmented the same way as the image?

Thank you!