ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

[Datawhale学习打卡]tianchi-intel-PaddleOCR

2021-07-15 23:58:13  阅读:292  来源: 互联网

标签:ch intel -- PaddleOCR det v2.0 infer 打卡 ppocr


[Datawhale学习打卡]tianchi-intel-PaddleOCR

一:安装Paddle-gpu

  1. 30系列安培架构显卡,建议装CUDA11.2版本的Paddle:
    python -m pip install paddlepaddle-gpu==2.1.1.post112 -f https://paddlepaddle.org.cn/whl/mkl/stable.html
  2. 验证安装
    import paddle
    paddle.utils.run_check()

二:克隆项目
3. git clone https://gitee.com/coggle/tianchi-intel-PaddleOCR
4. cd tianchi-intel-PaddleOCR
5. 不使用run.sh自带的一键训练
6. python3 down_image.py 下载图片
7.
7.1 mkdir inference
7.2 cd inference
7.3 下载三个步骤模型权重
wget https://paddleocr.bj.bcebos.com/dygraph_v2.0/ch/ch_ppocr_server_v2.0_det_infer.tar
wget https://paddleocr.bj.bcebos.com/dygraph_v2.0/ch/ch_ppocr_mobile_v2.0_cls_infer.tar
wget https://paddleocr.bj.bcebos.com/dygraph_v2.0/ch/ch_ppocr_server_v2.0_rec_infer.tar
7.4 解压在当前文件夹下
tar -xf ch_ppocr_server_v2.0_rec_infer.tar
tar -xf ch_ppocr_server_v2.0_det_infer.tar
tar -xf ch_ppocr_mobile_v2.0_cls_infer.tar
8. 验证推理 (注意当前位置)
python tools/infer/predict_system.py --image_dir="./1.jpg" --det_model_dir="./inference/ch_ppocr_server_v2.0_det_infer/" --rec_model_dir="./inference/ch_ppocr_server_v2.0_rec_infer/" --cls_model_dir=“./inference/ch_ppocr_mobile_v2.0_cls_infer/” --use_angle_cls=True --use_space_char=True

三: 迁移学习
9.
9.1 cd inference
9.2 wget https://paddleocr.bj.bcebos.com/dygraph_v2.0/ch/ch_ppocr_server_v2.0_det_train.tar
tar -xf ch_ppocr_server_v2.0_det_train.tar
9.3 windows修改模型文件加载线程数为0(训练集和验证集)
configs/det/ch_ppocr_v2.0/ch_det_res18_db_v2.0.yml
9.4 开始训练(配置文件修改。。10epoch收敛)
python tools/train.py -c configs/det/ch_ppocr_v2.0/ch_det_res18_db_v2.0.yml -o Global.pretrain_weights=./inference/ch_ppocr_server_v2.0_det_train/

四: 预测
10. 将模型导出
python tools/export_model.py -c configs/det/ch_ppocr_v2.0/ch_det_res18_db_v2.0.yml -o Global.pretrained_model=output/ch_db_res18/best_accuracy Global.save_inference_dir=output/ch_db_res18/

  1. 对测试集进行预测
    python tools/infer/predict_system_tianchi.py --image_dir="./doc/imgs/11.jpg" --det_model_dir=“output/ch_db_res18/” --rec_model_dir="./inference/ch_ppocr_server_v2.0_rec_infer/" --cls_model_dir=’./inference/ch_ppocr_mobile_v2.0_cls_infer/’ --use_angle_cls=True --use_space_char=True

  2. 将结果文件压缩
    zip -r submit.zip Xeon1OCR_round1_test*

AI-Studio项目地址:
https://aistudio.baidu.com/aistudio/projectdetail/2183888?shared=1

标签:ch,intel,--,PaddleOCR,det,v2.0,infer,打卡,ppocr
来源: https://blog.csdn.net/qq_38102943/article/details/118771460

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有