How do I get an RGB image after testing

[zhaoyucong]

During the training, I only trained Y channel, but did not synthesize it into RGB at the end of the training. Could you give me some Suggestions on how to change it

[cjsoft]

here's pseudo code

tmp = image.converttoYCbCr()
tmp['Y'] = vdsr(tmp['y'])
tmp = tmp.convertoRGB()
imshow(tmp)