nju-jet / sr_framework Goto Github PK

I am trying to save the SR image. I just directly use the func cv2.imwrite(path, tensor2np(sr)), but I get the wrong output image with wired color.
I just wonder how can I save the super-resolution image?

I try to modify the VDSR.yaml and DRRN.yaml, set the train_Y: True, but it seems I cannot do that, it will raise the error：

IndexError: too many indices for array: array is 2-dimensional, but 3 were indexed

What should I do?