From ec0e2082c0f2b13274402f59c2be73eaf7126d2d Mon Sep 17 00:00:00 2001
From: liuyebo <1515783401@qq.com>
Date: Tue, 27 Aug 2024 17:11:19 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BD=BF=E7=94=A8gpu=E8=B0=83=E7=94=A8?=
 =?UTF-8?q?=E6=96=87=E6=A1=A3=E6=A3=80=E6=B5=8B=E6=A8=A1=E5=9E=8B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 docker-compose.yml           | 2 +-
 paddle_detection/detector.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/docker-compose.yml b/docker-compose.yml
index 4700cd9..7b2180a 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -1,6 +1,6 @@
 x-env:
   &template
-  image: fcb_photo_review:1.13.2
+  image: fcb_photo_review:1.13.3
   restart: always
 
 services:
diff --git a/paddle_detection/detector.py b/paddle_detection/detector.py
index 94d48b6..db72a7c 100644
--- a/paddle_detection/detector.py
+++ b/paddle_detection/detector.py
@@ -33,7 +33,7 @@ def detect_image(img_path):
     onnx_file = "model/object_det_model/ppyoloe_plus_crn_x_80e_coco_w_nms.onnx"
     infer_cfg = "model/object_det_model/infer_cfg.yml"
     # load predictor
-    predictor = InferenceSession(onnx_file)
+    predictor = InferenceSession(onnx_file, providers=["CUDAExecutionProvider"])
     # load infer config
     infer_config = PredictConfig(infer_cfg)