未验证 提交 8a3769d8 编写于 作者: B Bubbliiiing 提交者: GitHub

Update yolo_training.py

上级 3e563fad
......@@ -498,7 +498,7 @@ class Generator(object):
img = np.array(img,dtype = np.float32)
inputs.append(np.transpose(img/255.0,(2,0,1)))
targets.append(y)
targets.append(np.array(y))
if len(targets) == self.batch_size:
tmp_inp = np.array(inputs)
tmp_targets = np.array(targets)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册