temp_bboxes get wrong dimension info

2026-09-24 23:33:08 +08:00 · 2021-12-30 14:45:16 +08:00
parent 547224099b
commit 725e845485
1 changed files with 1 additions and 1 deletions
@@ -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