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)