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

5 years ago · 9ae868364a
parent 8e5c66579b
commit 9ae868364a
1 changed files with 1 additions and 1 deletions
--- a/utils/datasets.py
+++ b/utils/datasets.py
@ -457,7 +457,7 @@ class LoadImagesAndLabels(Dataset):  # for training/testing
                    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)