diff --git a/configs/det/det_r50_vd_sast_totaltext.yml b/configs/det/det_r50_vd_sast_totaltext.yml index 257ecf2490bdde6280cf4b20bb66f2457b4b833b..a92f1b6e539b9f78d2edc705cd9cda0fb6522c28 100755 --- a/configs/det/det_r50_vd_sast_totaltext.yml +++ b/configs/det/det_r50_vd_sast_totaltext.yml @@ -60,7 +60,8 @@ Metric: Train: dataset: name: SimpleDataSet - label_file_path: [./train_data/art_latin_icdar_14pt/train_no_tt_test/train_label_json.txt, ./train_data/total_text_icdar_14pt/train_label_json.txt] + data_dir: ./train_data/ + label_file_list: [./train_data/art_latin_icdar_14pt/train_no_tt_test/train_label_json.txt, ./train_data/total_text_icdar_14pt/train_label_json.txt] data_ratio_list: [0.5, 0.5] transforms: - DecodeImage: # load image diff --git a/ppocr/utils/save_load.py b/ppocr/utils/save_load.py index af2de054de3656638fee8d4328765c21b4deaea4..02814d6208aba7ddfa6eac338229502b18b535da 100644 --- a/ppocr/utils/save_load.py +++ b/ppocr/utils/save_load.py @@ -102,7 +102,6 @@ def init_model(config, model, logger, optimizer=None, lr_scheduler=None): best_model_dict = states_dict.get('best_model_dict', {}) if 'epoch' in states_dict: best_model_dict['start_epoch'] = states_dict['epoch'] + 1 - best_model_dict['start_epoch'] = best_model_dict['best_epoch'] + 1 logger.info("resume from {}".format(checkpoints)) elif pretrained_model: