mirrors
/
PaddleOCR
mirror of https://github.com/PaddlePaddle/PaddleOCR.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

modify rec saved model dir

pull/5/head
LDOUBLEV 2020-05-13 16:51:36 +08:00
parent 8007a95528
commit 9126cb6fb9
9 changed files with 9 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
configs/rec/rec_mv3_none_bilstm_ctc.yml
View File

@ -4,7 +4,7 @@ Global:
epoch_num: 72 epoch_num: 72
log_smooth_window: 20 log_smooth_window: 20
print_batch_step: 10 print_batch_step: 10
save_model_dir: output/rec save_model_dir: output/rec_CRNN
save_epoch_step: 3 save_epoch_step: 3
eval_batch_step: 2000 eval_batch_step: 2000
train_batch_size_per_card: 256 train_batch_size_per_card: 256

2
configs/rec/rec_mv3_none_none_ctc.yml
View File

@ -4,7 +4,7 @@ Global:
epoch_num: 72 epoch_num: 72
log_smooth_window: 20 log_smooth_window: 20
print_batch_step: 10 print_batch_step: 10
save_model_dir: output/rec save_model_dir: output/rec_Rosetta
save_epoch_step: 3 save_epoch_step: 3
eval_batch_step: 2000 eval_batch_step: 2000
train_batch_size_per_card: 256 train_batch_size_per_card: 256

2
configs/rec/rec_mv3_tps_bilstm_attn.yml
View File

@ -4,7 +4,7 @@ Global:
epoch_num: 72 epoch_num: 72
log_smooth_window: 20 log_smooth_window: 20
print_batch_step: 10 print_batch_step: 10
save_model_dir: output/rec save_model_dir: output/rec_RARE
save_epoch_step: 3 save_epoch_step: 3
eval_batch_step: 2000 eval_batch_step: 2000
train_batch_size_per_card: 256 train_batch_size_per_card: 256

2
configs/rec/rec_mv3_tps_bilstm_ctc.yml
View File

@ -4,7 +4,7 @@ Global:
epoch_num: 72 epoch_num: 72
log_smooth_window: 20 log_smooth_window: 20
print_batch_step: 10 print_batch_step: 10
save_model_dir: output/rec save_model_dir: output/rec_STARNet
save_epoch_step: 3 save_epoch_step: 3
eval_batch_step: 2000 eval_batch_step: 2000
train_batch_size_per_card: 256 train_batch_size_per_card: 256

2
configs/rec/rec_r34_vd_none_bilstm_ctc.yml
View File

@ -4,7 +4,7 @@ Global:
epoch_num: 72 epoch_num: 72
log_smooth_window: 20 log_smooth_window: 20
print_batch_step: 10 print_batch_step: 10
save_model_dir: output/rec save_model_dir: output/rec_CRNN
save_epoch_step: 3 save_epoch_step: 3
eval_batch_step: 2000 eval_batch_step: 2000
train_batch_size_per_card: 256 train_batch_size_per_card: 256

2
configs/rec/rec_r34_vd_none_none_ctc.yml
View File

@ -4,7 +4,7 @@ Global:
epoch_num: 72 epoch_num: 72
log_smooth_window: 20 log_smooth_window: 20
print_batch_step: 10 print_batch_step: 10
save_model_dir: output/rec save_model_dir: output/rec_Rosetta
save_epoch_step: 3 save_epoch_step: 3
eval_batch_step: 2000 eval_batch_step: 2000
train_batch_size_per_card: 256 train_batch_size_per_card: 256

2
configs/rec/rec_r34_vd_tps_bilstm_attn.yml
View File

@ -4,7 +4,7 @@ Global:
epoch_num: 72 epoch_num: 72
log_smooth_window: 20 log_smooth_window: 20
print_batch_step: 10 print_batch_step: 10
save_model_dir: output/rec save_model_dir: output/rec_RARE
save_epoch_step: 3 save_epoch_step: 3
eval_batch_step: 2000 eval_batch_step: 2000
train_batch_size_per_card: 256 train_batch_size_per_card: 256

2
configs/rec/rec_r34_vd_tps_bilstm_ctc.yml
View File

@ -4,7 +4,7 @@ Global:
epoch_num: 72 epoch_num: 72
log_smooth_window: 20 log_smooth_window: 20
print_batch_step: 10 print_batch_step: 10
save_model_dir: output/rec save_model_dir: output/rec_STARNet
save_epoch_step: 3 save_epoch_step: 3
eval_batch_step: 2000 eval_batch_step: 2000
train_batch_size_per_card: 256 train_batch_size_per_card: 256

4
tools/program.py
View File

@ -284,9 +284,7 @@ def train_eval_rec_run(config, exe, train_info_dict, eval_info_dict):
eval_batch_step = config['Global']['eval_batch_step'] eval_batch_step = config['Global']['eval_batch_step']
save_epoch_step = config['Global']['save_epoch_step'] save_epoch_step = config['Global']['save_epoch_step']
save_model_dir = config['Global']['save_model_dir'] save_model_dir = config['Global']['save_model_dir']
if save_model_dir[-1] == "/": if not os.path.exists(save_model_dir):
save_model_dir = save_model_dir[:-1]
if not os.path.exists(save_model_dir + config['Global']['algorithm']):
os.makedirs(save_model_dir) os.makedirs(save_model_dir)
train_stats = TrainingStats(log_smooth_window, ['loss', 'acc']) train_stats = TrainingStats(log_smooth_window, ['loss', 'acc'])
best_eval_acc = -1 best_eval_acc = -1