From ae41d5855a2150e9c394a0e0fb5f0abebc9e5635 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Thu, 2 May 2019 15:47:27 +0200
Subject: [PATCH] updates

---
 utils/datasets.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/utils/datasets.py b/utils/datasets.py
index 490687ab..7561b629 100755
--- a/utils/datasets.py
+++ b/utils/datasets.py
@@ -179,7 +179,7 @@ class LoadImagesAndLabels(Dataset):  # for training/testing
         #    self.imgs = [cv2.imread(img_files[i]) for i in range(n)]
 
         # Preload labels (required for weighted CE training)
-        self.labels = [np.array([])] * n
+        self.labels = [np.zeros((0, 5))] * n
         iter = tqdm(self.label_files, desc='Reading labels') if n > 1000 else self.label_files
         for i, file in enumerate(iter):
             try: