Fix numpy.bool deprecation problem (#9404)
parent
15d7ef5765
commit
9c22490f95
|
@ -64,7 +64,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=bool)
|
||||||
|
|
||||||
data['polys'] = boxes
|
data['polys'] = boxes
|
||||||
data['texts'] = txts
|
data['texts'] = txts
|
||||||
|
@ -218,7 +218,7 @@ class E2ELabelEncodeTest(BaseRecLabelEncode):
|
||||||
else:
|
else:
|
||||||
txt_tags.append(False)
|
txt_tags.append(False)
|
||||||
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=bool)
|
||||||
data['polys'] = boxes
|
data['polys'] = boxes
|
||||||
data['ignore_tags'] = txt_tags
|
data['ignore_tags'] = txt_tags
|
||||||
temp_texts = []
|
temp_texts = []
|
||||||
|
@ -254,7 +254,7 @@ class E2ELabelEncodeTrain(object):
|
||||||
else:
|
else:
|
||||||
txt_tags.append(False)
|
txt_tags.append(False)
|
||||||
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=bool)
|
||||||
|
|
||||||
data['polys'] = boxes
|
data['polys'] = boxes
|
||||||
data['texts'] = txts
|
data['texts'] = txts
|
||||||
|
|
Loading…
Reference in New Issue