Merge pull request #5130 from Bourne-M/patch-2

temp_bboxes get wrong dimension info
This commit is contained in:
Double_V
2021-12-30 15:44:44 +08:00
committed by GitHub
+1 -1
View File
@@ -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