diff --git a/configs/e2e/e2e_r50_vd_pg.yml b/configs/e2e/e2e_r50_vd_pg.yml index 4c9e538707e96288c0739fb01e9006ccc609999a..2f1fab2ee15eba1cd3abccff7552aa0fd43621d1 100644 --- a/configs/e2e/e2e_r50_vd_pg.yml +++ b/configs/e2e/e2e_r50_vd_pg.yml @@ -63,7 +63,7 @@ PostProcess: Metric: name: E2EMetric - mode: A # A or B + mode: A # two ways for eval, A: label from txt, B: label from gt_mat gt_mat_dir: ./train_data/total_text/gt # the dir of gt_mat character_dict_path: ppocr/utils/ic15_dict.txt main_indicator: f_score_e2e diff --git a/ppocr/data/imaug/label_ops.py b/ppocr/data/imaug/label_ops.py index 0e3e048c745f99a6381d68f518696c3ef6142ced..0225001f296f570995a88a53b5366996db3cc12b 100644 --- a/ppocr/data/imaug/label_ops.py +++ b/ppocr/data/imaug/label_ops.py @@ -217,7 +217,7 @@ class E2ELabelEncode_test(BaseRecLabelEncode): boxes = np.array(boxes, dtype=np.float32) txt_tags = np.array(txt_tags, dtype=np.bool) data['polys'] = boxes - data['ignore_tags'] = txt_tags + data['tags'] = txt_tags temp_texts = [] for text in txts: text = text.lower() @@ -255,7 +255,7 @@ class E2ELabelEncode_train(object): data['polys'] = boxes data['texts'] = txts - data['ignore_tags'] = txt_tags + data['tags'] = txt_tags return data diff --git a/ppocr/data/imaug/pg_process.py b/ppocr/data/imaug/pg_process.py index 53031064c019ddce00c7546f898ac67a7f0459f9..0012466bce710753bc41107b946fffda5e9d47b1 100644 --- a/ppocr/data/imaug/pg_process.py +++ b/ppocr/data/imaug/pg_process.py @@ -750,7 +750,7 @@ class PGProcessTrain(object): input_size = 512 im = data['image'] text_polys = data['polys'] - text_tags = data['ignore_tags'] + text_tags = data['tags'] text_strs = data['texts'] h, w, _ = im.shape text_polys, text_tags, hv_tags = self.check_and_validate_polys(