Wrong result and code

Hi:
Thank you for the great work!
I downloaded the code,but it may be wrong in util.py function load_image,and I don't load your checkpoint because it cannot find the key in the checkpoint.
Could you help me to confirm this?

Best