From d5fc54f95a6f7fff996f1144c679292ee140889b Mon Sep 17 00:00:00 2001 From: dyning Date: Fri, 15 May 2020 23:15:22 +0800 Subject: [PATCH] Update inference.md --- doc/inference.md | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/doc/inference.md b/doc/inference.md index 9d3724c2..7ae15d11 100644 --- a/doc/inference.md +++ b/doc/inference.md @@ -28,6 +28,11 @@ python3 tools/infer/predict_det.py --image_dir="./doc/imgs/2.jpg" --det_model_di python3 tools/infer/predict_det.py --image_dir="./doc/imgs/2.jpg" --det_model_dir="./inference/det/" --det_max_side_len=1200 ``` +# 如果想使用CPU进行预测，执行命令如下 +``` +python3 tools/infer/predict_det.py --image_dir="./doc/imgs/2.jpg" --det_model_dir="./inference/det/" --use_gpu=False +``` + ### 2.DB文本检测模型推理 首先将DB文本检测训练过程中保存的模型，转换成inference model。以基于Resnet50_vd骨干网络，在ICDAR2015英文数据集训练的模型为例（[模型下载地址](https://paddleocr.bj.bcebos.com/det_r50_vd_db.tar))，可以使用如下命令进行转换： -- GitLab