How to access crude data at loading for data augmentation

[KanielDatz]

Hello, I'm trying to augment my data by adding some random objects to the images, stiches, staples and more elements that might be on x-ray images.
I wrote a nice function that goes into the augmentation pipeline but when I tried saving some of the images I found out that the functions receives already edited images , they seem saturated and their size differs from the original.

I tried to find the code that loads the images and add the augmentation code there but I couldn't figure it out.
searched here:

nnunetv2\training\dataloading

But didn't understand how the actual images are loaded, just found dictionaries representing the dataset.

Hope to find some help, Thanks,
Daniel