[Datawhale学习打卡]tianchi-intel-PaddleOCR
2021/7/16 6:08:04
本文主要是介绍[Datawhale学习打卡]tianchi-intel-PaddleOCR,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!
[Datawhale学习打卡]tianchi-intel-PaddleOCR
一:安装Paddle-gpu
- 30系列安培架构显卡,建议装CUDA11.2版本的Paddle:
python -m pip install paddlepaddle-gpu==2.1.1.post112 -f https://paddlepaddle.org.cn/whl/mkl/stable.html - 验证安装
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/
-
对测试集进行预测
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 -
将结果文件压缩
zip -r submit.zip Xeon1OCR_round1_test*
AI-Studio项目地址:
https://aistudio.baidu.com/aistudio/projectdetail/2183888?shared=1
这篇关于[Datawhale学习打卡]tianchi-intel-PaddleOCR的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!
- 2024-11-23增量更新怎么做?-icode9专业技术文章分享
- 2024-11-23压缩包加密方案有哪些?-icode9专业技术文章分享
- 2024-11-23用shell怎么写一个开机时自动同步远程仓库的代码?-icode9专业技术文章分享
- 2024-11-23webman可以同步自己的仓库吗?-icode9专业技术文章分享
- 2024-11-23在 Webman 中怎么判断是否有某命令进程正在运行?-icode9专业技术文章分享
- 2024-11-23如何重置new Swiper?-icode9专业技术文章分享
- 2024-11-23oss直传有什么好处?-icode9专业技术文章分享
- 2024-11-23如何将oss直传封装成一个组件在其他页面调用时都可以使用?-icode9专业技术文章分享
- 2024-11-23怎么使用laravel 11在代码里获取路由列表?-icode9专业技术文章分享
- 2024-11-22怎么实现ansible playbook 备份代码中命名包含时间戳功能?-icode9专业技术文章分享