the value should be a list to unpacking (#688)
parent
8e5c66579b
commit
9ae868364a
|
@ -457,7 +457,7 @@ class LoadImagesAndLabels(Dataset): # for training/testing
|
||||||
l = np.zeros((0, 5), dtype=np.float32)
|
l = np.zeros((0, 5), dtype=np.float32)
|
||||||
x[img] = [l, shape]
|
x[img] = [l, shape]
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
x[img] = None
|
x[img] = [None, None]
|
||||||
print('WARNING: %s: %s' % (img, e))
|
print('WARNING: %s: %s' % (img, e))
|
||||||
|
|
||||||
x['hash'] = get_hash(self.label_files + self.img_files)
|
x['hash'] = get_hash(self.label_files + self.img_files)
|
||||||
|
|
Loading…
Reference in New Issue