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}])