search

ZFTurbo / volumentations

Library for 3D augmentations
MIT License
220 stars 35 forks source link

multi input/modality augmentation #31

Open jizhang02 opened 9 months ago

jizhang02 commented 9 months ago

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 commented 3 months ago

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