Wrong result and code

[yiifanLu]

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