Solved by Caruso in post #2 
              
              
                Hi, 
for me your augmentation looks good. You could visualize your input data by denormalizing it first, converting it from [B, C, H, W] to [B, H, W, C] using torch.permute(), maybe min-max it to be between 0-255, then convert it to a numpy array and e.g. show it with matplotlib imshow(). 
Greetings…