Test dataset

prismer_model.py

@@ -69,7 +69,9 @@ def compute_md5(image_path: str) -> str:
 
 def run_experts(image_path: str) -> Tuple[str, Tuple[str, ...]]:
     im_name = compute_md5(image_path)
-    out_path = submodule_dir / 'helpers' / 'images' / f'{im_name}.jpg'
+    image_dir = submodule_dir / 'helpers' / 'images'
+    out_path = image_dir / f'{im_name}.jpg'
+    image_dir.mkdir(parents=True, exist_ok=True)
     keys = ['depth', 'edge', 'normal', 'seg_coco', 'obj_detection', 'ocr_detection']
     results = [pathlib.Path('prismer/helpers/labels') / key / f'helpers/images/{im_name}.png' for key in keys]
     results_pretty = [pathlib.Path('prismer/helpers/labels') / key / f'helpers/images/{im_name}_p.png' for key in keys]