updates

2019-07-30 17:51:19 +02:00
parent 65abb1c82f
commit b11bb91804
2 changed files with 8 additions and 6 deletions
@@ -213,7 +213,7 @@ class LoadImagesAndLabels(Dataset):  # for training/testing
        self.imgs = [None] * n
        self.labels = [None] * n
        preload_labels = False
-        if preload_labels:
+        if preload_labels or image_weights:
            self.labels = [np.zeros((0, 5))] * n
            extract_bounding_boxes = False
            for i, file in enumerate(tqdm(self.label_files, desc='Reading labels') if n > 10 else self.label_files):