WeiYi95

Try this: python gen.py --checkpoint=./model_dir/HAN/model.ckpt-* Replace * with your checkpoint No.

Change example_img = Image.new(mode, (canvas_size * 2, canvas_size), (255, 255, 255)) to example_img = Image.new(mode, (canvas_size * 2, canvas_size), "#FFFFFF")