multi input/modality augmentation

[jizhang02]

Hello,

I would like to ask if the input images are CT and PET, as well as mask, how should I use the code below? Thank you.

# with mask
data = {'image': img, 'mask': lbl}
aug_data = aug(**data)
img, lbl = aug_data['image'], aug_data['mask']

[SergioRodLla]

Hi!

I was about to try to do the same. If your images are in NIfTI format I would suggest to read them with nibabel, convert then to numpy arrays, and apply the function of your code. Any ideas are welcome.

Regards,
Sergio