Saving color images

[nicesonnday]

In the 'train.py' code, the input color images are resized and transposed by (1,2,0).

Why did you transposed the original input images when you save the "data" including original images if opt.mode=='test' and opt.save_prediction==True. ?

[zhenpeiyang]

Hi @nicesonnday , if you asking about this line, the transpose is just for converting the [channel, height, width] format to [height, width, channel] format.