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

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

Browse Source
This commit is contained in:
liuyebo
2024-08-28 13:53:12 +08:00
parent 155a2d3d28
commit 8085c75207
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docker-compose.yml
View File

@@ -1,6 +1,6 @@
x-env: x-env:
&template &template
image: fcb_photo_review:1.13.4 image: fcb_photo_review:1.13.5
restart: always restart: always
services: services:

4
paddle_detection/__init__.py
View File

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