mirror of
https://github.com/PaddlePaddle/PaddleOCR.git
synced 2026-09-24 23:33:08 +08:00
Merge pull request #5130 from Bourne-M/patch-2
temp_bboxes get wrong dimension info
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user