经测试，cpu推理比gpu快，修改文档检测的配置

2024-08-28 13:53:12 +08:00
parent 155a2d3d28
commit 8085c75207
2 changed files with 3 additions and 3 deletions
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -1,6 +1,6 @@
 x-env:
  &template
-  image: fcb_photo_review:1.13.4
+  image: fcb_photo_review:1.13.5
  restart: always

 services:
--- a/paddle_detection/init.py
+++ b/paddle_detection/init.py
@@ -1,4 +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}])
+PADDLE_DET = InferenceSession("model/object_det_model/ppyoloe_plus_crn_x_80e_coco_w_nms.onnx",
+                              providers=["CPUExecutionProvider"], provider_options=[{"device_id": 0}])