commit
3ec9e56994
|
@ -128,4 +128,4 @@ Eval:
|
|||
drop_last: False
|
||||
batch_size_per_card: 1 # must be 1
|
||||
num_workers: 8
|
||||
use_shared_memory: False
|
||||
use_shared_memory: False
|
||||
|
|
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue