Spaces:
Runtime error
Runtime error
Remove parallelization
Browse files- prismer_model.py +9 -4
prismer_model.py
CHANGED
@@ -63,10 +63,15 @@ def run_experts(image_path: str) -> tuple[str | None, ...]:
|
|
63 |
out_path = image_dir / 'image.jpg'
|
64 |
cv2.imwrite(out_path.as_posix(), cv2.imread(image_path))
|
65 |
|
66 |
-
expert_names = ['edge', 'normal', 'objdet', 'ocrdet', 'segmentation']
|
67 |
-
run_expert('depth')
|
68 |
-
with concurrent.futures.ProcessPoolExecutor() as executor:
|
69 |
-
|
|
|
|
|
|
|
|
|
|
|
70 |
|
71 |
keys = ['depth', 'edge', 'normal', 'seg_coco', 'obj_detection', 'ocr_detection']
|
72 |
results = [pathlib.Path('prismer/helpers/labels') / key / 'helpers/images/image.png' for key in keys]
|
|
|
63 |
out_path = image_dir / 'image.jpg'
|
64 |
cv2.imwrite(out_path.as_posix(), cv2.imread(image_path))
|
65 |
|
66 |
+
# expert_names = ['edge', 'normal', 'objdet', 'ocrdet', 'segmentation']
|
67 |
+
# run_expert('depth')
|
68 |
+
# with concurrent.futures.ProcessPoolExecutor() as executor:
|
69 |
+
# executor.map(run_expert, expert_names)
|
70 |
+
|
71 |
+
# no parallelization just to be safe
|
72 |
+
expert_names = ['depth', 'edge', 'normal', 'objdet', 'ocrdet', 'segmentation']
|
73 |
+
for exp in expert_names:
|
74 |
+
run_expert(exp)
|
75 |
|
76 |
keys = ['depth', 'edge', 'normal', 'seg_coco', 'obj_detection', 'ocr_detection']
|
77 |
results = [pathlib.Path('prismer/helpers/labels') / key / 'helpers/images/image.png' for key in keys]
|