fix char dict

上级 f1a62304
...@@ -77,6 +77,7 @@ class ArgsParser(ArgumentParser): ...@@ -77,6 +77,7 @@ class ArgsParser(ArgumentParser):
global_config['Global']['save_model_dir'] = './output/rec_{}_lite'.format(type[0]) global_config['Global']['save_model_dir'] = './output/rec_{}_lite'.format(type[0])
global_config['Train']['dataset']['label_file_list'] = ["train_data/{}_train.txt".format(type[0])] global_config['Train']['dataset']['label_file_list'] = ["train_data/{}_train.txt".format(type[0])]
global_config['Eval']['dataset']['label_file_list'] = ["train_data/{}_val.txt".format(type[0])] global_config['Eval']['dataset']['label_file_list'] = ["train_data/{}_val.txt".format(type[0])]
global_config['Global']['character_type'] = type[0]
assert( assert(
os.path.isfile(os.path.join(project_path,global_config['Global']['character_dict_path'])) os.path.isfile(os.path.join(project_path,global_config['Global']['character_dict_path']))
),"Loss default dictionary file {}_dict.txt.You can download it from \ ),"Loss default dictionary file {}_dict.txt.You can download it from \
......
...@@ -17,7 +17,7 @@ Global: ...@@ -17,7 +17,7 @@ Global:
# for data or label process # for data or label process
character_dict_path: character_dict_path:
# Set the language of training, if set, select the default dictionary file # Set the language of training, if set, select the default dictionary file
character_type: ch character_type:
max_text_length: 25 max_text_length: 25
infer_mode: False infer_mode: False
use_space_char: True use_space_char: True
......
...@@ -137,7 +137,7 @@ z ...@@ -137,7 +137,7 @@ z
©
ª
«
¬
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册