优化文档检测

2024-08-28 10:53:37 +08:00
parent e794002d28
commit d0f3f783ce
3 changed files with 7 additions and 6 deletions
--- a/paddle_detection/init.py
+++ b/paddle_detection/init.py
@@ -0,0 +1,4 @@
+from onnxruntime import InferenceSession
+
+PADDLE_DET = InferenceSession("model/object_det_model/ppyoloe_plus_crn_l_80e_coco_w_nms.onnx",
+                              providers=["CUDAExecutionProvider"], provider_options=[{"device_id": 0}])