Merge pull request #26 from huihuilong/feature/ui-infer-api

fix:指定onnxruntime SessionOptions的intra_op_num_threads，解决centos7上pthre…
Meituan-Dianping · Apr 1, 2022 · b05108d · b05108d
2 parents 080081d + 27b4e86
commit b05108d
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)