Hi,
I have created a class that extends DataSet to load images for a segmentation task, so one input and one output. Every time the method getitem is called, this class performs the necessary operations for data augmentation on both the input and the output, and it works perfectly.
However, when I use this class with PyTorch DataLoader, the input transformation do not match with the output transformations. My bet is that to perform the same operations, I have to get/set the states of random operations, and the DataLoader does the same.
How can I fix it?
Regards,