Skip to content

GitLab

Unverified Commit 725e8454 authored by yfzhou's avatar yfzhou Committed by GitHub
Browse files

temp_bboxes get wrong dimension info

parent 54722409
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
ppocr/data/imaug/label_ops.py
View file @ 725e8454
...@@ -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% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment