PaddleOCR使用笔记-模型推理和预测

最新推荐文章于 2024-07-23 23:17:41 发布

快乐的小小程序猿

最新推荐文章于 2024-07-23 23:17:41 发布

阅读量1.6k

点赞数 1

分类专栏：深度学习 PaddleOCR 文章标签：深度学习 PaddleOCR Linux 模型推理

本文链接：https://blog.csdn.net/xu_benjamin/article/details/120088513

版权

深度学习同时被 2 个专栏收录

45 篇文章 3 订阅

订阅专栏

PaddleOCR

23 篇文章 2 订阅

订阅专栏

这篇博客详细记录了在Linux环境下使用PaddleOCR的develop版本进行模型安装、转换以及推理预测的全过程，包括检测、识别和方向分类模型的训练与应用。通过export_model.py将模型转化为推理模型，然后利用predict_det.py、predict_rec.py和predict_cls.py进行预测，最后展示了如何进行检测识别模型和方向分类模型的串联预测。

摘要由CSDN通过智能技术生成

PaddleOCR使用笔记

Linux环境下

PaddleOCR develop版本安装及使用（第二次尝试）(成功)

检测模型转推理模型

python3 tools/export_model.py -c configs/det/det_mv3_east.yml -o Global.checkpoints=output/det_east_mv3/best_accuracy Global.save_inference_dir=./inference/det_mv3_east

在这里插入图片描述

识别模型转推理模型

python3 tools/export_model.py -c configs/rec/multi_languages/rec_en_number_lite_train.yml -o Global.checkpoints=output/rec_en_number_lite_crown/best_accuracy Global.save_inference_dir=./inference/rec_en_number_lite_crown

python3 tools/export_model.py -c configs/rec/multi_languages/rec_en_number_lite_train2.yml -o Global.checkpoints=output/rec_aug_en_number_lite_crown/best_accuracy Global.save_inference_dir=./inference/rec_aug_en_number_lite_crown

在这里插入图片描述

方向分类模型转推理模型

python3 tools/export_model.py -c configs/cls/cls_mv3.yml -o Global.checkpoints=output/cls_mv3/best_accuracy Global.save_inference_dir=./inference/cls_crown

在这里插入图片描述

检测推理模型预测

python3 tools/infer/predict_det.py --det_algorithm="EAST" --image_dir="./doc/imgs_crown_test/" --det_model_dir="./inference/det_mv3_east/"

在这里插入图片描述

识别推理模型预测

python3 tools/infer/predict_rec.py --image_dir="./doc/imgs_crown_crop_test/" --rec_model_dir="./inference/rec_en_number_lite_crown" --rec_char_type="ch" --rec_char_dict_path="ppocr/utils/dict/en_dict.txt" --use_space_char=False

python3 tools/infer/predict_rec.py --image_dir="./doc/aug_rec_crown_test/" --rec_model_dir="./inference/rec_aug_en_number_lite_crown" --rec_char_type="ch" --rec_char_dict_path="ppocr/utils/dict/en_dict.txt" --use_space_char=False

在这里插入图片描述

方向分类推理模型预测

python3 tools/infer/predict_cls.py --image_dir="./doc/cls_crown_test/" --cls_model_dir="./inference/cls_crown/"

在这里插入图片描述

检测识别推理模型串联预测

python3 tools/infer/predict_system.py --image_dir="./doc/imgs_crown_test/" --det_model_dir="./inference/det_mv3_east/" --rec_model_dir="./inference/rec_en_number_lite_crown/" --use_angle_cls=false --use_space_char=False --rec_char_type="ch" --det_algorithm="EAST" --rec_char_dict_path='ppocr/utils/dict/en_dict.txt'

python3 tools/infer/predict_system.py --image_dir="./train_data/crown_dataset/det_crown/test/" --det_model_dir="./inference/det_mv3_east/" --rec_model_dir="./inference/rec_en_number_lite_crown/" --use_angle_cls=false --use_space_char=False --rec_char_type="ch" --det_algorithm="EAST" --rec_char_dict_path='ppocr/utils/dict/en_dict.txt'

python3 tools/infer/predict_system.py --image_dir="./train_data/crown_dataset/det_crown/test/" --det_model_dir="./inference/det_mv3_east/" --rec_model_dir="./inference/rec_aug_en_number_lite_crown/" --use_angle_cls=false --use_space_char=False --rec_char_type="ch" --det_algorithm="EAST" --rec_char_dict_path='ppocr/utils/dict/en_dict.txt'

在这里插入图片描述

检测识别方向分类推理模型串联预测

python3 tools/infer/predict_system.py --image_dir="./doc/det_rec_cls_data_test/" --det_model_dir="./inference/det_mv3_east/" --rec_model_dir="./inference/rec_aug_en_number_lite_crown/" --cls_model_dir="./inference/cls_crown/" --use_angle_cls=true --use_space_char=False --rec_char_type="ch" --det_algorithm="EAST" --rec_char_dict_path='ppocr/utils/dict/en_dict.txt'