shikunl commited on
Commit
d1c3a3a
β€’
1 Parent(s): 50b6c28

Remove parallelization

Browse files
Files changed (1) hide show
  1. 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
- executor.map(run_expert, expert_names)
 
 
 
 
 
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]