优化文档检测的结果

This commit is contained in:
2024-08-29 10:21:08 +08:00
parent 68046582d6
commit df324c2be8
3 changed files with 27 additions and 6 deletions

View File

@@ -1,4 +1,4 @@
from onnxruntime import InferenceSession
PADDLE_DET = InferenceSession("model/object_det_model/ppyoloe_plus_crn_x_80e_coco_w_nms.onnx",
PADDLE_DET = InferenceSession("model/object_det_model/ppyoloe_plus_crn_l_80e_coco_w_nms.onnx",
providers=["CPUExecutionProvider"], provider_options=[{"device_id": 0}])