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)