About color distortion in the restored images

[wrqcodedoge]

Hello, thank you for your excellent work!

I noticed a significant color distortion in the restored images when running the super-resolution task. The restored image is shown on the right.
Do you know how to solve this issue?
I look forward to your response.

[sjames40]

you need to use the clear_color(output) which is the function from util.img_utils import clear_color