config.md 4.5 KB
Newer Older
K
Khanh Tran 已提交
1
# Optional parameters list
T
tink2123 已提交
2

K
Khanh Tran 已提交
3
The following list can be viewed via `--help`
T
tink2123 已提交
4

K
Khanh Tran 已提交
5
|         FLAG             |     Supported script    |        Use        |      Defaults       |         Note         |
T
tink2123 已提交
6
| :----------------------: | :------------: | :---------------: | :--------------: | :-----------------: |
K
Khanh Tran 已提交
7 8
|          -c              |      ALL       |  Specify configuration file |  None  |  **Please refer to the parameter introduction for configuration file usage** |
|          -o              |      ALL       |  Set the parameter in the configuration file  |  None  |  Configuration using -o has higher priority than the configuration file selected with -c. E.g: `-o Global.use_gpu=false`  |  
T
tink2123 已提交
9 10


K
Khanh Tran 已提交
11
## Introduction to Global Parameters of Configuration File 
T
tink2123 已提交
12

K
Khanh Tran 已提交
13
Take `rec_chinese_lite_train.yml` as an example
T
tink2123 已提交
14 15


K
Khanh Tran 已提交
16
|         Parameter             |            Use                |      Default       |            Note            |
T
tink2123 已提交
17
| :----------------------: |  :---------------------:   | :--------------:  |   :--------------------:   |
K
Khanh Tran 已提交
18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42
|      algorithm           |    Select algorithm to use                    |  Synchronize with configuration file   |     For selecting model, please refer to the supported model [list](https://github.com/PaddlePaddle/PaddleOCR/blob/develop/README.md) |
|      use_gpu             |    Set using GPU or not            |       true        |                \                 |
|      epoch_num           |    Maximum training epoch number             |       3000        |                \                 |
|      log_smooth_window   |    Sliding window size            |       20          |                \                 |
|      print_batch_step    |    Set print log interval         |       10          |                \                 |
|      save_model_dir      |    Set model save path        |  output/{model_name}  |                \                 |
|      save_epoch_step     |    Set model save interval        |       3           |                \                 |
|      eval_batch_step     |    Set the model evaluation interval        |       2000        |                \                 |
|train_batch_size_per_card |  Set the batch size during training   |         256         |                \                 |
| test_batch_size_per_card |  Set the batch size during testing    |         256         |                \                 |
|      image_shape         |    Set input image size        |   [3, 32, 100]    |                \                 |
|      max_text_length     |    Set the maximum text length        |       25          |                \                 |
|      character_type      |    Set character type            |       ch          |    en/ch, the default dict will be used for en, and the custom dict will be used for ch|
|      character_dict_path |    Set dictionary path            |  ./ppocr/utils/ic15_dict.txt  |    \                 |
|      loss_type           |    Set loss type              |       ctc         |    Supports two types of loss: ctc / attention |
|      reader_yml          |    Set the reader configuration file          |  ./configs/rec/rec_icdar15_reader.yml  |  \          |
|      pretrain_weights    |    Load pre-trained model path      |  ./pretrain_models/CRNN/best_accuracy  |  \          |
|      checkpoints         |    Load saved model path            |       None        |    Used to load saved parameters to continue training after interruption |
|      save_inference_dir  |   path to save model for inference |          None        |   Use to save inference model |

## Introduction Reader parameters of Configuration file

Take `rec_chinese_reader.yml` as an example:

|         Parameter             |            Use                |      Default       |            Note            |
T
tink2123 已提交
43
| :----------------------: |  :---------------------:   | :--------------:  |   :--------------------:   |
K
Khanh Tran 已提交
44 45 46 47 48
|      reader_function     |    Select data reading method        |  ppocr.data.rec.dataset_traversal,SimpleReader  | Support two data reading methods: SimpleReader / LMDBReader  |
|      num_workers             |    Set the number of data reading threads            |       8        |                \                 |
|      img_set_dir          |    Image folder path             |       ./train_data        |                \                 |
|      label_file_path      |    Groundtruth file path           |       ./train_data/rec_gt_train.txt| \    |
|      infer_img            |    Result folder path     |       ./infer_img | \|
T
tink2123 已提交
49