From aae39ca894300cea594b47f426020ad41b683371 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 7 Nov 2019 14:46:16 -0800 Subject: [PATCH] updates --- utils/datasets.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/utils/datasets.py b/utils/datasets.py index fe783651..8ccccd39 100755 --- a/utils/datasets.py +++ b/utils/datasets.py @@ -579,6 +579,7 @@ def load_mosaic(self, index): with open(label_path, 'r') as f: x = np.array([x.split() for x in f.read().splitlines()], dtype=np.float32) + labels = [] if x.size > 0: # Normalized xywh to pixel xyxy format labels = x.copy() @@ -586,8 +587,8 @@ def load_mosaic(self, index): labels[:, 2] = h * (x[:, 2] - x[:, 4] / 2) + padh labels[:, 3] = w * (x[:, 1] + x[:, 3] / 2) + padw labels[:, 4] = h * (x[:, 2] + x[:, 4] / 2) + padh - labels4.append(labels) + if len(labels4): labels4 = np.concatenate(labels4, 0)