显存不足，回滚

Revert "使用gpu调用文档检测模型" This reverts commit ec0e2082c0.
2024-08-27 17:17:45 +08:00
parent ec0e2082c0
commit e794002d28
2 changed files with 2 additions and 2 deletions
--- 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:
--- 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)