Echo/fcb_photo_review
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

使用gpu调用文档检测模型

Browse Source
This commit is contained in:
liuyebo
2024-08-27 17:11:19 +08:00
parent 4009d7eb81
commit ec0e2082c0
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docker-compose.yml
View File

@@ -1,6 +1,6 @@
x-env: x-env:
&template &template
image: fcb_photo_review:1.13.2 image: fcb_photo_review:1.13.3
restart: always restart: always
services: services:

2
paddle_detection/detector.py
View File

@@ -33,7 +33,7 @@ def detect_image(img_path):
onnx_file = "model/object_det_model/ppyoloe_plus_crn_x_80e_coco_w_nms.onnx" onnx_file = "model/object_det_model/ppyoloe_plus_crn_x_80e_coco_w_nms.onnx"
infer_cfg = "model/object_det_model/infer_cfg.yml" infer_cfg = "model/object_det_model/infer_cfg.yml"
# load predictor # load predictor
predictor = InferenceSession(onnx_file) predictor = InferenceSession(onnx_file, providers=["CUDAExecutionProvider"])
# load infer config # load infer config
infer_config = PredictConfig(infer_cfg) infer_config = PredictConfig(infer_cfg)