How to view the dataset after data augmentation

[casekevin]

Hello

I am curious about how nnUNet decides the augmentation to the dataset, so is there ways we could extract the dataset after the data augmentation before the initialization of the network?

Thank you!

[FabianIsensee]

data augmentation is done on the fly during training, so there is no augmented dataset you can look at. That you can do is save the samples that the network sees in run_iteration

[casekevin]

Thank you so much!

[jmrichusc]

Hello,

Could you explain how to save the samples that the network sees in run_iteration for nnUNetTrainerV2?

Also, how would one view the augmented dataset in nnUNetTrainerV2_DA5?

[FabianIsensee]

You need to code this yourself. You have already identified the right location. Just add some code that exports the batches