diff --git a/nets/yolo_training.py b/nets/yolo_training.py index b84b4d0515c8500290f5061f9ad874f592cb9f99..263c45b4da982d125b3587470d3a77f244ba7adb 100644 --- a/nets/yolo_training.py +++ b/nets/yolo_training.py @@ -372,7 +372,7 @@ class Generator(object): targets.append(np.array(y,dtype = np.float32)) if len(targets) == self.batch_size: tmp_inp = np.array(inputs) - tmp_targets = np.array(targets) + tmp_targets = targets inputs = [] targets = [] yield tmp_inp, tmp_targets