From 725e8454855a496c7dbc9ed285dfe9962e88e2e8 Mon Sep 17 00:00:00 2001 From: yfzhou Date: Thu, 30 Dec 2021 14:45:16 +0800 Subject: [PATCH] temp_bboxes get wrong dimension info --- ppocr/data/imaug/label_ops.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ppocr/data/imaug/label_ops.py b/ppocr/data/imaug/label_ops.py index fc14fdbc..f83255b7 100644 --- a/ppocr/data/imaug/label_ops.py +++ b/ppocr/data/imaug/label_ops.py @@ -344,7 +344,7 @@ class KieLabelEncode(object): max_num = 300 temp_bboxes = np.zeros([max_num, 4]) h, _ = bboxes.shape - temp_bboxes[:h, :h] = bboxes + temp_bboxes[:h, :] = bboxes temp_relations = np.zeros([max_num, max_num, 5]) temp_relations[:h, :h, :] = relations -- GitLab