Xu Ma commited on
Commit
b1d3cdc
1 Parent(s): e10c15b

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -1
main.py CHANGED
@@ -371,7 +371,8 @@ def main_func(target, experiment, cfg_arg):
371
  device = pydiffvg.get_device()
372
 
373
  print(f"debug: {cfg.target}")
374
- gt = np.array(PIL.Image.open(cfg.target))
 
375
  print(f"Input image shape is: {gt.shape}")
376
  if len(gt.shape) == 2:
377
  print("Converting the gray-scale image to RGB.")
 
371
  device = pydiffvg.get_device()
372
 
373
  print(f"debug: {cfg.target}")
374
+ # gt = np.array(PIL.Image.open(cfg.target))
375
+ gt = cfg.target
376
  print(f"Input image shape is: {gt.shape}")
377
  if len(gt.shape) == 2:
378
  print("Converting the gray-scale image to RGB.")