未验证 提交 5e03500f 编写于 作者: B Bubbliiiing 提交者: GitHub

Update yolo_training.py

上级 a5694268
...@@ -515,7 +515,7 @@ class Generator(object): ...@@ -515,7 +515,7 @@ class Generator(object):
if train: if train:
img,y=self.get_random_data(annotation_line, self.image_size[0:2]) img,y=self.get_random_data(annotation_line, self.image_size[0:2])
else: else:
img,y=self.get_random_data(annotation_line, self.image_size[0:2], False) img,y=self.get_random_data(annotation_line, self.image_size[0:2], random=False)
if len(y)!=0: if len(y)!=0:
boxes = np.array(y[:,:4],dtype=np.float32) boxes = np.array(y[:,:4],dtype=np.float32)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册