diff --git a/ppocr/data/imaug/label_ops.py b/ppocr/data/imaug/label_ops.py index fc14fdbcf13a61b591d9ea6c2535aefe6e437ec6..f83255b732f5990de6a99d1149bab77d682c85b3 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