From e794002d2887a97505c5baa1bc56ce3e3cd383ae Mon Sep 17 00:00:00 2001
From: liuyebo <1515783401@qq.com>
Date: Tue, 27 Aug 2024 17:17:45 +0800
Subject: [PATCH] =?UTF-8?q?=E6=98=BE=E5=AD=98=E4=B8=8D=E8=B6=B3=EF=BC=8C?=
 =?UTF-8?q?=E5=9B=9E=E6=BB=9A=20Revert=20"=E4=BD=BF=E7=94=A8gpu=E8=B0=83?=
 =?UTF-8?q?=E7=94=A8=E6=96=87=E6=A1=A3=E6=A3=80=E6=B5=8B=E6=A8=A1=E5=9E=8B?=
 =?UTF-8?q?"?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

This reverts commit ec0e2082c0f2b13274402f59c2be73eaf7126d2d.
---
 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 7b2180a..4700cd9 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -1,6 +1,6 @@
 x-env:
   &template
-  image: fcb_photo_review:1.13.3
+  image: fcb_photo_review:1.13.2
   restart: always
 
 services:
diff --git a/paddle_detection/detector.py b/paddle_detection/detector.py
index db72a7c..94d48b6 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, providers=["CUDAExecutionProvider"])
+    predictor = InferenceSession(onnx_file)
     # load infer config
     infer_config = PredictConfig(infer_cfg)