Commit e6e49cf0 authored by wanglch's avatar wanglch
Browse files

Update README.md

parent 1e5c5a4a
...@@ -23,7 +23,7 @@ CRNN(卷积循环神经网络)是一种用于光学字符识别(OCR)的 ...@@ -23,7 +23,7 @@ CRNN(卷积循环神经网络)是一种用于光学字符识别(OCR)的
## 环境配置 ## 环境配置
### Docker(方法一) ### Docker(方法一)
推荐使用docker方式运行, 此处提供[光源](https://www.sourcefind.cn/#/service-details)拉取docker镜像的地址与使用步骤 推荐使用docker方式运行, 此处提供[光源](https://www.sourcefind.cn/#/service-list)拉取docker镜像的地址与使用步骤
``` ```
docker pull image.sourcefind.cn:5000/dcu/admin/base/paddlepaddle:3.0.0-py3.10-dtk24.04.3-ubuntu20.04 docker pull image.sourcefind.cn:5000/dcu/admin/base/paddlepaddle:3.0.0-py3.10-dtk24.04.3-ubuntu20.04
...@@ -92,13 +92,14 @@ export LD_LIBRARY_PATH=/opt/dtk-24.04.3/lib:/opt/dtk-24.04.3/cuda/extras/CUPTI/l ...@@ -92,13 +92,14 @@ export LD_LIBRARY_PATH=/opt/dtk-24.04.3/lib:/opt/dtk-24.04.3/cuda/extras/CUPTI/l
训练集应有如下文件结构: 训练集应有如下文件结构:
``` ```
|-train_data |-train_data
|-rec |-icdar2015
|- rec_gt_train.txt |-rec
|- train |- rec_gt_train.txt
|- word_001.png |- train
|- word_002.jpg |- word_001.png
|- word_003.jpg |- word_002.jpg
| ... |- word_003.jpg
| ...
``` ```
## 训练 ## 训练
...@@ -123,7 +124,7 @@ configs/rec/rec_r34_vd_none_bilstm_ctc.yml ...@@ -123,7 +124,7 @@ configs/rec/rec_r34_vd_none_bilstm_ctc.yml
``` ```
### 运行训练文件 ### 运行训练文件
``` ```
CUDA_VISIBLE_DEVICES=1 python3 tools/train.py -c ./PaddleOCR-main/configs/rec/rec_r34_vd_none_bilstm_ctc.yml -o Global.pretrained_model=./PaddleOCR-main/pretrain_models/rec_mv3_none_bilstm_ctc_v2.0_train/best_accuracy CUDA_VISIBLE_DEVICES=1 python3 tools/train.py -c ./configs/rec/rec_r34_vd_none_bilstm_ctc.yml -o Global.pretrained_model=./pretrain_models/rec_r34_vd_none_bilstm_ctc_v2.0_train/best_accuracy
``` ```
...@@ -135,7 +136,7 @@ CUDA_VISIBLE_DEVICES=1 python3 tools/train.py -c ./PaddleOCR-main/configs/rec/re ...@@ -135,7 +136,7 @@ CUDA_VISIBLE_DEVICES=1 python3 tools/train.py -c ./PaddleOCR-main/configs/rec/re
``` ```
# 预测使用的配置文件必须与训练一致 # 预测使用的配置文件必须与训练一致
python3 tools/infer_rec.py -c configs/rec/rec_r34_vd_none_bilstm_ctc.yml -o Global.pretrained_model='./PaddleOCR-main/pretrain_models/rec_r34_vd_none_bilstm_ctc_v2.0_train/best_accuracy' Global.infer_img='./PaddleOCR-main/docs/images/ppocrv4.png' python3 tools/infer_rec.py -c configs/rec/rec_r34_vd_none_bilstm_ctc.yml -o Global.pretrained_model='./pretrain_models/rec_r34_vd_none_bilstm_ctc_v2.0_train/best_accuracy' Global.infer_img='./docs/images/ppocrv4.png'
``` ```
...@@ -164,7 +165,7 @@ python3 tools/infer_rec.py -c configs/rec/rec_r34_vd_none_bilstm_ctc.yml -o Glob ...@@ -164,7 +165,7 @@ python3 tools/infer_rec.py -c configs/rec/rec_r34_vd_none_bilstm_ctc.yml -o Glob
## 预训练权重 ## 预训练权重
CRNN PaddlePaddle下载地址为:[rec_r34_vd_none_bilstm_ctc.yml](https://paddleocr.bj.bcebos.com/dygraph_v2.0/en/rec_r34_vd_none_bilstm_ctc_v2.0_train.tar) CRNN PaddlePaddle下载地址为:[rec_r34_vd_none_bilstm_ctc](https://paddleocr.bj.bcebos.com/dygraph_v2.0/en/rec_r34_vd_none_bilstm_ctc_v2.0_train.tar)
## 源码仓库及问题反馈 ## 源码仓库及问题反馈
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment