Test dataset

prismer_model.py

@@ -83,9 +83,7 @@ def run_experts(image_path: str) -> Tuple[str, Tuple[str, ...]]:
     with open('prismer/configs/experts.yaml', 'w') as yaml_file:
         yaml.dump(config, yaml_file, default_flow_style=False)
 
-    print(out_path.as_posix())
     if not os.path.exists(out_paths[0]):
-        print(out_path.as_posix())
         cv2.imwrite(out_path.as_posix(), cv2.imread(image_path))
 
         # paralleled inference