From 39955558b8d2ff6c956ed36917c6c33f6aeaa6f6 Mon Sep 17 00:00:00 2001 From: WenmuZhou Date: Wed, 16 Dec 2020 19:41:44 +0800 Subject: [PATCH] fix bugs --- configs/det/det_r50_vd_sast_totaltext.yml | 3 ++- ppocr/utils/save_load.py | 1 - 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/configs/det/det_r50_vd_sast_totaltext.yml b/configs/det/det_r50_vd_sast_totaltext.yml index 257ecf24..a92f1b6e 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 af2de054..02814d62 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: -- GitLab