Ah, sorry, it should have been [:, :, ::-1] but that doesn’t work in PyTorch but you need .flip(2) (after that, the three image channels should be switched), you can use matplotib’s pyplot.imshow to see whether it looks OK.