lelouedec:

But the transpose is changing the order of channels
At the end I have a H W C but values inside at not the same.
And ToTensor uses HxWxC array . so … I don’t understand


Maybe you should also change the order of means and stds