Merge remote-tracking branch 'origin/master'

Meituan-Dianping · Apr 2, 2022 · 1b33616 · 1b33616
2 parents 56561d1 + b05108d
commit 1b33616
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/service/image_infer.py b/service/image_infer.py
@@ -14,7 +14,10 @@ def __init__(self, model_path):
         self.cls_thresh = 0.5
         self.nms_thresh = 0.2
         self.model_path = model_path
-        self.model_session = onnxruntime.InferenceSession(self.model_path)
+        so = onnxruntime.SessionOptions()
+        # 配置intra_op_num_threads为CPU核数时最佳
+        so.intra_op_num_threads = 4
+        self.model_session = onnxruntime.InferenceSession(self.model_path, sess_options=so)
 
     def ui_infer(self, image_path):
         origin_img = cv2.imread(image_path)