From b2d91baee1023e9c9e906bb747d869e22fb468f4 Mon Sep 17 00:00:00 2001 From: Lovely-Pig <1607806186@qq.com> Date: Thu, 30 Dec 2021 00:46:37 +0800 Subject: [PATCH] Update customize.md --- doc/doc_ch/customize.md | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/doc/doc_ch/customize.md b/doc/doc_ch/customize.md index 5944bf08..65371739 100644 --- a/doc/doc_ch/customize.md +++ b/doc/doc_ch/customize.md @@ -5,17 +5,21 @@ ## step1:训练文本检测模型 PaddleOCR提供了EAST、DB两种文本检测算法,均支持MobileNetV3、ResNet50_vd两种骨干网络,根据需要选择相应的配置文件,启动训练。例如,训练使用MobileNetV3作为骨干网络的DB检测模型(即超轻量模型使用的配置): -``` + +```sh python3 tools/train.py -c configs/det/det_mv3_db.yml 2>&1 | tee det_db.log ``` + 更详细的数据准备和训练教程参考文档教程中[文本检测模型训练/评估/预测](./detection.md)。 ## step2:训练文本识别模型 PaddleOCR提供了CRNN、Rosetta、STAR-Net、RARE四种文本识别算法,均支持MobileNetV3、ResNet34_vd两种骨干网络,根据需要选择相应的配置文件,启动训练。例如,训练使用MobileNetV3作为骨干网络的CRNN识别模型(即超轻量模型使用的配置): -``` + +```sh python3 tools/train.py -c configs/rec/rec_chinese_lite_train.yml 2>&1 | tee rec_ch_lite.log ``` + 更详细的数据准备和训练教程参考文档教程中[文本识别模型训练/评估/预测](./recognition.md)。 ## step3:模型串联预测 @@ -24,7 +28,8 @@ PaddleOCR提供了检测和识别模型的串联工具,可以将训练好的 在执行预测时,需要通过参数image_dir指定单张图像或者图像集合的路径、参数det_model_dir指定检测inference模型的路径和参数rec_model_dir指定识别inference模型的路径。可视化识别结果默认保存到 ./inference_results 文件夹里面。 -``` +```sh python3 tools/infer/predict_system.py --image_dir="./doc/imgs/11.jpg" --det_model_dir="./inference/det/" --rec_model_dir="./inference/rec/" ``` + 更多的文本检测、识别串联推理使用方式请参考文档教程中的[基于预测引擎推理](./inference.md)。 -- GitLab