tkianai commited on
Commit
9ae8683
1 Parent(s): 8e5c665

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

Browse files
Files changed (1) hide show
  1. utils/datasets.py +1 -1
utils/datasets.py CHANGED
@@ -457,7 +457,7 @@ class LoadImagesAndLabels(Dataset): # for training/testing
457
  l = np.zeros((0, 5), dtype=np.float32)
458
  x[img] = [l, shape]
459
  except Exception as e:
460
- x[img] = None
461
  print('WARNING: %s: %s' % (img, e))
462
 
463
  x['hash'] = get_hash(self.label_files + self.img_files)
 
457
  l = np.zeros((0, 5), dtype=np.float32)
458
  x[img] = [l, shape]
459
  except Exception as e:
460
+ x[img] = [None, None]
461
  print('WARNING: %s: %s' % (img, e))
462
 
463
  x['hash'] = get_hash(self.label_files + self.img_files)