Update main.py

main.py

@@ -372,7 +372,7 @@ def main_func(target, experiment, cfg_arg):
 
     print(f"debug: {cfg.target}")
     # gt = np.array(PIL.Image.open(cfg.target))
-    gt = cfg.target
+    gt = np.array(cfg.target)
     print(f"Input image shape is: {gt.shape}")
     if len(gt.shape) == 2:
         print("Converting the gray-scale image to RGB.")