From 84b1488f8d616bdd6f1136da5f8c962bca23461d Mon Sep 17 00:00:00 2001
From: Bubbliiiing <47347516+bubbliiiing@users.noreply.github.com>
Date: Wed, 9 Sep 2020 15:31:32 +0800
Subject: [PATCH] Update yolo_training.py

---
 nets/yolo_training.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/nets/yolo_training.py b/nets/yolo_training.py
index b84b4d0..263c45b 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
-- 
GitLab