修正模型地址

2024-09-26 16:56:48 +08:00
parent 603b027ca6
commit b5aba0418b
2 changed files with 3 additions and 4 deletions
--- a/services/paddle_services/paddle_detection/init.py
+++ b/services/paddle_services/paddle_detection/init.py
@@ -2,7 +2,6 @@ import os

 from onnxruntime import InferenceSession

-MODEL_DIR = os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))),
-                         'model', 'object_det_model')
+MODEL_DIR = os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), 'model', 'object_det_model')
 PADDLE_DET = InferenceSession(os.path.join(MODEL_DIR, 'ppyoloe_plus_crn_l_80e_coco_w_nms.onnx'),
                              providers=['CPUExecutionProvider'], provider_options=[{'device_id': 0}])