提交 e9a60ec7 编写于 作者: L LDOUBLEV

fix config

上级 8e679d38
...@@ -18,8 +18,7 @@ Global: ...@@ -18,8 +18,7 @@ Global:
infer_img: doc/imgs_en/img_10.jpg infer_img: doc/imgs_en/img_10.jpg
save_res_path: ./checkpoints/det_db/predicts_db.txt save_res_path: ./checkpoints/det_db/predicts_db.txt
distributed: true distributed: true
student_pretrained_model: ./pretrain_model/MobileNetV3_large_x0_5_pretrained
teacher_pretrained_model: ./inference/ch_ppocr_resnet50/student
Architecture: Architecture:
model_type: det model_type: det
algorithm: DB algorithm: DB
...@@ -57,6 +56,7 @@ NetWorks: ...@@ -57,6 +56,7 @@ NetWorks:
k: 50 k: 50
kernel_size: [7, 2, 2] kernel_size: [7, 2, 2]
Teacher: Teacher:
pretrained: ./inference/ch_ppocr_resnet50/student
model_type: det model_type: det
algorithm: DB algorithm: DB
Backbone: Backbone:
...@@ -128,23 +128,10 @@ Metric: ...@@ -128,23 +128,10 @@ Metric:
Train: Train:
dataset: dataset:
name: SimpleDataSet name: SimpleDataSet
data_dir: ../chinese/ data_dir: ./train_data/icdar2015/text_localization/
label_file_list: label_file_list:
- ../chinese/zhongce_training_fix_1.6k.txt - ./train_data/icdar2015/text_localization/train_icdar2015_label.txt
- ../chinese/label_train_all_f4_part2.txt ratio_list: [1.0]
- ../chinese/label_train_all_f4_part4.txt
- ../chinese/label_train_all_f4_part5_no_born-digit_ifly5000.txt
- ../chinese/synch_en_ch_largeword_ppocr_2W.txt
- ../chinese/pubtabnet_ppocr_5W_new.txt
- ../chinese/label_synth_sheet_1W.txt
ratio_list:
- 1
- 0.1
- 0.1
- 0.4
- 0.05
- 0.04
- 0.05
transforms: transforms:
- DecodeImage: - DecodeImage:
img_mode: BGR img_mode: BGR
...@@ -205,9 +192,9 @@ Train: ...@@ -205,9 +192,9 @@ Train:
Eval: Eval:
dataset: dataset:
name: SimpleDataSet name: SimpleDataSet
data_dir: ../chinese/test_set/ data_dir: ./train_data/icdar2015/text_localization/
label_file_list: label_file_list:
- ../chinese/test_set/Label_refine_v8.txt - ./train_data/icdar2015/text_localization/test_icdar2015_label.txt
transforms: transforms:
- DecodeImage: - DecodeImage:
img_mode: BGR img_mode: BGR
......
...@@ -18,8 +18,7 @@ Global: ...@@ -18,8 +18,7 @@ Global:
infer_img: doc/imgs_en/img_10.jpg infer_img: doc/imgs_en/img_10.jpg
save_res_path: ./checkpoints/det_db/predicts_db.txt save_res_path: ./checkpoints/det_db/predicts_db.txt
distributed: true distributed: true
student_pretrained_model: ./pretrain_model/MobileNetV3_large_x0_5_pretrained
teacher_pretrained_model: ./inference/ch_ppocr_resnet50/student
Architecture: Architecture:
model_type: det model_type: det
algorithm: DB algorithm: DB
...@@ -67,23 +66,10 @@ Metric: ...@@ -67,23 +66,10 @@ Metric:
Train: Train:
dataset: dataset:
name: SimpleDataSet name: SimpleDataSet
data_dir: ../chinese/ data_dir: ./train_data/icdar2015/text_localization/
label_file_list: label_file_list:
- ../chinese/zhongce_training_fix_1.6k.txt - ./train_data/icdar2015/text_localization/train_icdar2015_label.txt
- ../chinese/label_train_all_f4_part2.txt ratio_list: [1.0]
- ../chinese/label_train_all_f4_part4.txt
- ../chinese/label_train_all_f4_part5_no_born-digit_ifly5000.txt
- ../chinese/synch_en_ch_largeword_ppocr_2W.txt
- ../chinese/pubtabnet_ppocr_5W_new.txt
- ../chinese/label_synth_sheet_1W.txt
ratio_list:
- 1
- 0.1
- 0.1
- 0.4
- 0.05
- 0.04
- 0.05
transforms: transforms:
- DecodeImage: - DecodeImage:
img_mode: BGR img_mode: BGR
...@@ -144,9 +130,9 @@ Train: ...@@ -144,9 +130,9 @@ Train:
Eval: Eval:
dataset: dataset:
name: SimpleDataSet name: SimpleDataSet
data_dir: ../chinese/test_set/ data_dir: ./train_data/icdar2015/text_localization/
label_file_list: label_file_list:
- ../chinese/test_set/Label_refine_v8.txt - ./train_data/icdar2015/text_localization/test_icdar2015_label.txt
transforms: transforms:
- DecodeImage: - DecodeImage:
img_mode: BGR img_mode: BGR
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册