Update inference.md

d5fc54f9 · dyning · GitHub · 27e6d883 · d5fc54f9
Unverified Commit d5fc54f9 authored May 15, 2020 by dyning Committed by GitHub May 15, 2020
Show whitespace changes
Inline Side-by-side

Showing with 5 additions and 0 deletions

doc/inference.md doc/inference.md +5 -0

No files found.
--- 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))，可以使用如下命令进行转换：