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

This commit is contained in:
tkianai authored and GitHub committed 2020-08-10 11:23:48 -07:00
1 parent 8e5c66579b
commit 9ae868364a
1 file changed
+1 -1
+1 -1
View File
@@ -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)