修复

cyclegan.py

@@ -66,7 +66,7 @@ class CYCLEGAN(object):
         #---------------------------------------------------------#
         #   添加上batch_size维度
         #---------------------------------------------------------#
-        image_data = np.expand_dims(np.transpose(preprocess_input(np.array(image_data, dtype='float32')), (2, 0, 1)), 0)
+        image_data = np.expand_dims(np.transpose(preprocess_input(np.array(image, dtype='float32')), (2, 0, 1)), 0)
         
         with torch.no_grad():
             images = torch.from_numpy(image_data)