shikunl commited on
Commit
78492f4
Β·
1 Parent(s): 864efc4

Test dataset

Browse files
Files changed (1) hide show
  1. prismer_model.py +2 -0
prismer_model.py CHANGED
@@ -81,7 +81,9 @@ def run_experts(image_path: str) -> Tuple[str, Tuple[str, ...]]:
81
  with open('prismer/configs/experts.yaml', 'w') as yaml_file:
82
  yaml.dump(config, yaml_file, default_flow_style=False)
83
 
 
84
  if not os.path.exists(out_paths[0]):
 
85
  cv2.imwrite(out_path.as_posix(), cv2.imread(image_path))
86
 
87
  # paralleled inference
 
81
  with open('prismer/configs/experts.yaml', 'w') as yaml_file:
82
  yaml.dump(config, yaml_file, default_flow_style=False)
83
 
84
+ print(out_path.as_posix())
85
  if not os.path.exists(out_paths[0]):
86
+ print(out_path.as_posix())
87
  cv2.imwrite(out_path.as_posix(), cv2.imread(image_path))
88
 
89
  # paralleled inference