未验证 提交 84b1488f 编写于 作者: B Bubbliiiing 提交者: GitHub

Update yolo_training.py

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