forgetest5
/
PaddleOCR
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

return None when no boxes found

This commit is contained in:
LDOUBLEV 2021-08-03 06:55:21 +00:00
parent 51bd3a6bdf
commit c620b9394f
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
ppocr/data/imaug/label_ops.py
View File

@ -53,9 +53,11 @@ class DetLabelEncode(object):
txt_tags.append(True)
else:
txt_tags.append(False)
if len(boxes) == 0:
return None
boxes = self.expand_points_num(boxes)
#boxes = np.array(boxes, dtype=np.float32)
#txt_tags = np.array(txt_tags, dtype=np.bool)
boxes = np.array(boxes, dtype=np.float32)
txt_tags = np.array(txt_tags, dtype=np.bool)
data['polys'] = boxes
data['texts'] = txts