nnUNet icon indicating copy to clipboard operation
nnUNet copied to clipboard

Data augmentation transform - tr_transforms - plot

Open Chandhinii-Techstudio opened this issue 1 year ago • 2 comments

Hi Team, There are list of data augmentation transform in nnunetTrainer.py. I would like to apply each augmentation transform like spatial transform, gaussian noise transform etc on a particular image for testing purpose and check how the transform changes the image by visualizing via plot.

Kindly do guide me on how to implement and check them.

My data is nifti nibabel image 3d and segment is 3d with 2 foreground labels and background zero

Chandhinii-Techstudio avatar Dec 14 '23 04:12 Chandhinii-Techstudio

You have to rebuild your dataloader. In nnU-Net dataloading happens through batchgenerators. So you just copy the dict with the transforms that you use, pass it to the dataloader and do whatever you like.

TaWald avatar Dec 19 '23 10:12 TaWald

@Chandhinii-Techstudio Does the dataloading /visualizing still pose an issue to you?

TaWald avatar Jan 24 '24 13:01 TaWald