diff --git a/doc/inference.md b/doc/inference.md index 9d3724c2738e009d645cadded29bf15c743bc2e8..7ae15d1177768147a3f4f61a1e9d1574cf317c23 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)),可以使用如下命令进行转换: