From 8085c75207396515e63043a7f3632fde1881a08e Mon Sep 17 00:00:00 2001 From: liuyebo <1515783401@qq.com> Date: Wed, 28 Aug 2024 13:53:12 +0800 Subject: [PATCH] =?UTF-8?q?=E7=BB=8F=E6=B5=8B=E8=AF=95=EF=BC=8Ccpu?= =?UTF-8?q?=E6=8E=A8=E7=90=86=E6=AF=94gpu=E5=BF=AB=EF=BC=8C=E4=BF=AE?= =?UTF-8?q?=E6=94=B9=E6=96=87=E6=A1=A3=E6=A3=80=E6=B5=8B=E7=9A=84=E9=85=8D?= =?UTF-8?q?=E7=BD=AE?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- docker-compose.yml | 2 +- paddle_detection/__init__.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/docker-compose.yml b/docker-compose.yml index 0147748..61dc8ce 100644 --- 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: diff --git a/paddle_detection/__init__.py b/paddle_detection/__init__.py index ba7922b..83cffe4 100644 --- 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}])