[bug report] Images rotated by 90 degrees

[ncble]

I have noticed that the images generated by the environment are inconsistent with the other features. In fact, due to the 'channel first' convention in Pytorch, the images should be transposed using np.transpose(0,3,1,2) instead of np.transpose(0,3,2,1); and np.transpose(2,0,1) instead of np.transpose(2,1,0). Several scripts need to be corrected: (both in the robotics-rl-srl and srl_zoo)

Scripts to be corrected:

• ./rl_baselines/evolution_strategies/cma_es.py: np.transpose(obs / 255.0, (0, 3, 1, 2))
• ./rl_baselines/evolution_strategies/cma_es.py: np.transpose(obs / 255.0, (0, 3, 1, 2))
• ./state_representation/models.py: np.transpose(observation, (0, 3, 2, 1))
• ./srl_zoo/plotting/representation_plot.py: np.transpose(image, (2, 1, 0))
• ./srl_zoo/preprocessing/utils.py: np.transpose(x, (2, 1, 0))
• ./srl_zoo/preprocessing/data_loader.py: im = th.tensor(im.reshape((1,) + im.shape).transpose(0, 3, 2, 1))