Skip to content

GitLab

Commit 5c858dc9 authored by dcuai's avatar dcuai
Browse files

Update label_ops.py—bool_

parent 8cb8274b
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
ppocr/data/imaug/label_ops.py
View file @ 5c858dc9
...@@ -62,7 +62,7 @@ class DetLabelEncode(object): ...@@ -62,7 +62,7 @@ class DetLabelEncode(object):
return None return None
boxes = self.expand_points_num(boxes) boxes = self.expand_points_num(boxes)
boxes = np.array(boxes, dtype=np.float32) boxes = np.array(boxes, dtype=np.float32)
txt_tags = np.array(txt_tags, dtype=np.bool) txt_tags = np.array(txt_tags, dtype=np.bool_)
data['polys'] = boxes data['polys'] = boxes
data['texts'] = txts data['texts'] = txts
...@@ -262,4 +262,4 @@ class MultiLabelEncode(BaseRecLabelEncode): ...@@ -262,4 +262,4 @@ class MultiLabelEncode(BaseRecLabelEncode):
data_out['label_sar'] = sar['label'] data_out['label_sar'] = sar['label']
data_out['length'] = ctc['length'] data_out['length'] = ctc['length']
return data_out return data_out
\ No newline at end of file
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