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

使用gpu运行文档检测模型

Browse Source
This commit is contained in:
liuyebo
2024-10-17 17:08:29 +08:00
parent cc53243647
commit 8e7745f1f6
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -4,4 +4,4 @@ from onnxruntime import InferenceSession
MODEL_DIR = os.path.join(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'), 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=['CUDAExecutionProvider'], provider_options=[{'device_id': 0}])