未验证 提交 1dad0ad3 编写于 作者: D Double_V 提交者: GitHub

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

temp_bboxes get wrong dimension info
...@@ -344,7 +344,7 @@ class KieLabelEncode(object): ...@@ -344,7 +344,7 @@ class KieLabelEncode(object):
max_num = 300 max_num = 300
temp_bboxes = np.zeros([max_num, 4]) temp_bboxes = np.zeros([max_num, 4])
h, _ = bboxes.shape h, _ = bboxes.shape
temp_bboxes[:h, :h] = bboxes temp_bboxes[:h, :] = bboxes
temp_relations = np.zeros([max_num, max_num, 5]) temp_relations = np.zeros([max_num, max_num, 5])
temp_relations[:h, :h, :] = relations temp_relations[:h, :h, :] = relations
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册