the value should be a list to unpacking (#688)

ultralytics · Aug 10, 2020 · 9ae8683 · 9ae8683
1 parent 8e5c665
commit 9ae8683
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/datasets.py b/utils/datasets.py
@@ -457,7 +457,7 @@ def cache_labels(self, path='labels.cache'):
                     l = np.zeros((0, 5), dtype=np.float32)
                 x[img] = [l, shape]
             except Exception as e:
-                x[img] = None
+                x[img] = [None, None]
                 print('WARNING: %s: %s' % (img, e))
 
         x['hash'] = get_hash(self.label_files + self.img_files)