diff --git a/configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_cml.yml b/configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_cml.yml index d8001197986e90d8fbb8f84ffb595948d6035c22..6eb3c9edd6778a07248a2b20a8a78b24b7318c2f 100644 --- a/configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_cml.yml +++ b/configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_cml.yml @@ -18,8 +18,7 @@ Global: infer_img: doc/imgs_en/img_10.jpg save_res_path: ./checkpoints/det_db/predicts_db.txt distributed: true - student_pretrained_model: ./pretrain_model/MobileNetV3_large_x0_5_pretrained - teacher_pretrained_model: ./inference/ch_ppocr_resnet50/student + Architecture: model_type: det algorithm: DB @@ -57,6 +56,7 @@ NetWorks: k: 50 kernel_size: [7, 2, 2] Teacher: + pretrained: ./inference/ch_ppocr_resnet50/student model_type: det algorithm: DB Backbone: @@ -128,23 +128,10 @@ Metric: Train: dataset: name: SimpleDataSet - data_dir: ../chinese/ + data_dir: ./train_data/icdar2015/text_localization/ label_file_list: - - ../chinese/zhongce_training_fix_1.6k.txt - - ../chinese/label_train_all_f4_part2.txt - - ../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 + - ./train_data/icdar2015/text_localization/train_icdar2015_label.txt + ratio_list: [1.0] transforms: - DecodeImage: img_mode: BGR @@ -205,9 +192,9 @@ Train: Eval: dataset: name: SimpleDataSet - data_dir: ../chinese/test_set/ + data_dir: ./train_data/icdar2015/text_localization/ label_file_list: - - ../chinese/test_set/Label_refine_v8.txt + - ./train_data/icdar2015/text_localization/test_icdar2015_label.txt transforms: - DecodeImage: img_mode: BGR diff --git a/configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_student.yml b/configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_student.yml index d081c26a75ae30e69146261b7d89e0b8fdc0b6ca..1d53d062dafeb414d109d48bc741dd2a1efbb63c 100644 --- a/configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_student.yml +++ b/configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_student.yml @@ -18,8 +18,7 @@ Global: infer_img: doc/imgs_en/img_10.jpg save_res_path: ./checkpoints/det_db/predicts_db.txt distributed: true - student_pretrained_model: ./pretrain_model/MobileNetV3_large_x0_5_pretrained - teacher_pretrained_model: ./inference/ch_ppocr_resnet50/student + Architecture: model_type: det algorithm: DB @@ -67,23 +66,10 @@ Metric: Train: dataset: name: SimpleDataSet - data_dir: ../chinese/ + data_dir: ./train_data/icdar2015/text_localization/ label_file_list: - - ../chinese/zhongce_training_fix_1.6k.txt - - ../chinese/label_train_all_f4_part2.txt - - ../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 + - ./train_data/icdar2015/text_localization/train_icdar2015_label.txt + ratio_list: [1.0] transforms: - DecodeImage: img_mode: BGR @@ -144,9 +130,9 @@ Train: Eval: dataset: name: SimpleDataSet - data_dir: ../chinese/test_set/ + data_dir: ./train_data/icdar2015/text_localization/ label_file_list: - - ../chinese/test_set/Label_refine_v8.txt + - ./train_data/icdar2015/text_localization/test_icdar2015_label.txt transforms: - DecodeImage: img_mode: BGR