修复

cyclegan.py

@@ -14,6 +14,10 @@ class CYCLEGAN(object):
         #   model_path指向logs文件夹下的权值文件
         #-----------------------------------------------#
         "model_path"        : 'model_data/G_model_B2A_last_epoch_weights.pth',
+        #-----------------------------------------------#
+        #   输入图像大小的设置
+        #-----------------------------------------------#
+        "input_shape"       : [112, 112],
         #-------------------------------#
         #   是否使用Cuda
         #   没有GPU可以设置成False