diff --git a/README.md b/README.md index 56e3f9930c692794cf9af867141849ae747bbe3b..49b7c808535a4b0152f38ac36f56c214bfb79950 100644 --- a/README.md +++ b/README.md @@ -36,7 +36,8 @@ PaddleOCR旨在打造一套丰富、领先、且实用的OCR工具库,助力 #### 2.inference模型下载 -*windows 环境下没有如果没有安装wget,下载模型时可将链接复制到浏览器中下载,并解压放置在相应目录下* +*windows 环境下如果没有安装wget,下载模型时可将链接复制到浏览器中下载,并解压放置在相应目录下* + #### (1)超轻量级中文OCR模型下载 ``` diff --git a/configs/rec/rec_benchmark_reader.yml b/configs/rec/rec_benchmark_reader.yml index ef4dd909c7d255cac484e2e87eadc69f6b32b9d2..524f2f68bac92ff6ffe3ff3b34e461d2adc81e41 100755 --- a/configs/rec/rec_benchmark_reader.yml +++ b/configs/rec/rec_benchmark_reader.yml @@ -1,7 +1,7 @@ TrainReader: reader_function: ppocr.data.rec.dataset_traversal,LMDBReader num_workers: 8 - lmdb_sets_dir: ./train_data/data_lmdb_release/evaluation/ + lmdb_sets_dir: ./train_data/data_lmdb_release/training/ EvalReader: reader_function: ppocr.data.rec.dataset_traversal,LMDBReader diff --git a/configs/rec/rec_icdar15_train.yml b/configs/rec/rec_icdar15_train.yml index b3b22985d46848f677a8cd000e88687984371223..934a94109cb304c5dd5e8db281f1fbf00d928e39 100755 --- a/configs/rec/rec_icdar15_train.yml +++ b/configs/rec/rec_icdar15_train.yml @@ -1,20 +1,20 @@ Global: algorithm: CRNN - use_gpu: false + use_gpu: true epoch_num: 1000 log_smooth_window: 20 print_batch_step: 10 save_model_dir: ./output/rec_CRNN save_epoch_step: 300 eval_batch_step: 500 - train_batch_size_per_card: 2 - test_batch_size_per_card: 2 + train_batch_size_per_card: 256 + test_batch_size_per_card: 256 image_shape: [3, 32, 100] max_text_length: 25 character_type: en loss_type: ctc reader_yml: ./configs/rec/rec_icdar15_reader.yml - pretrain_weights: + pretrain_weights: ./pretrain_models/rec_mv3_none_bilstm_ctc/best_accuracy checkpoints: save_inference_dir: infer_img: diff --git a/configs/rec/rec_mv3_tps_bilstm_attn.yml b/configs/rec/rec_mv3_tps_bilstm_attn.yml index a9dbce6a6c24c19ef07205cf73686a5357d89b7c..7fc4f6799459bf9fbcd25e1609aeca5e3fd12a74 100755 --- a/configs/rec/rec_mv3_tps_bilstm_attn.yml +++ b/configs/rec/rec_mv3_tps_bilstm_attn.yml @@ -1,14 +1,14 @@ Global: algorithm: RARE - use_gpu: false + use_gpu: true epoch_num: 72 log_smooth_window: 20 print_batch_step: 10 save_model_dir: output/rec_RARE save_epoch_step: 3 eval_batch_step: 2000 - train_batch_size_per_card: 2 - test_batch_size_per_card: 2 + train_batch_size_per_card: 256 + test_batch_size_per_card: 256 image_shape: [3, 32, 100] max_text_length: 25 character_type: en diff --git a/doc/detection.md b/doc/detection.md index 0fa32116e1c6cd53724b04e35486c31fefe6e4c4..ab12dfe945e7679a52a626e73ad7f8f587b0c58d 100644 --- a/doc/detection.md +++ b/doc/detection.md @@ -46,6 +46,9 @@ wget -P ./pretrain_models/ https://paddle-imagenet-models-name.bj.bcebos.com/Res ``` **启动训练** + +*如果您安装的是cpu版本,请将配置文件中的 `use_gpu` 字段修改为false* + ``` python3 tools/train.py -c configs/det/det_mv3_db.yml ``` diff --git a/doc/recognition.md b/doc/recognition.md index 7dd9ca7ecd56267c00db077eb290096dffa65de3..004487f92b972f728f7b093b31798a91b0ae0c56 100644 --- a/doc/recognition.md +++ b/doc/recognition.md @@ -113,6 +113,8 @@ tar -xf rec_mv3_none_bilstm_ctc.tar && rm -rf rec_mv3_none_bilstm_ctc.tar 开始训练: +*如果您安装的是cpu版本,请将配置文件中的 `use_gpu` 字段修改为false* + ``` # 设置PYTHONPATH路径 export PYTHONPATH=$PYTHONPATH:.