Echo/fcb_photo_review
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

修正模型地址

Browse Source
This commit is contained in:
liuyebo
2024-09-26 16:56:48 +08:00
parent 603b027ca6
commit b5aba0418b
2 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
services/paddle_services/doc_dewarp/__init__.py
View File

@@ -1,7 +1,7 @@
import os.path import os
from onnxruntime import InferenceSession from onnxruntime import InferenceSession
MODEL_PATH = os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))), MODEL_PATH = os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))),
'model', 'dewarp_model', 'doc_tr_pp.onnx') 'model', 'dewarp_model', 'doc_tr_pp.onnx')
DOC_TR = InferenceSession(MODEL_PATH, providers=['CUDAExecutionProvider'], provider_options=[{'device_id': 0}]) DOC_TR = InferenceSession(MODEL_PATH, providers=['CUDAExecutionProvider'], provider_options=[{'device_id': 0}])

3
services/paddle_services/paddle_detection/__init__.py
View File

@@ -2,7 +2,6 @@ import os
from onnxruntime import InferenceSession 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_DIR = os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), 'model', 'object_det_model')
'model', 'object_det_model')
PADDLE_DET = InferenceSession(os.path.join(MODEL_DIR, 'ppyoloe_plus_crn_l_80e_coco_w_nms.onnx'), PADDLE_DET = InferenceSession(os.path.join(MODEL_DIR, 'ppyoloe_plus_crn_l_80e_coco_w_nms.onnx'),
providers=['CPUExecutionProvider'], provider_options=[{'device_id': 0}]) providers=['CPUExecutionProvider'], provider_options=[{'device_id': 0}])