glenn-jocher commited on
Commit
0ab303b
1 Parent(s): 916bdb1

Remove `else:` from load_image() (#8692)

Browse files
Files changed (1) hide show
  1. utils/dataloaders.py +1 -2
utils/dataloaders.py CHANGED
@@ -682,8 +682,7 @@ class LoadImagesAndLabels(Dataset):
682
  interp = cv2.INTER_LINEAR if (self.augment or r > 1) else cv2.INTER_AREA
683
  im = cv2.resize(im, (int(w0 * r), int(h0 * r)), interpolation=interp)
684
  return im, (h0, w0), im.shape[:2] # im, hw_original, hw_resized
685
- else:
686
- return self.ims[i], self.im_hw0[i], self.im_hw[i] # im, hw_original, hw_resized
687
 
688
  def cache_images_to_disk(self, i):
689
  # Saves an image as an *.npy file for faster loading
 
682
  interp = cv2.INTER_LINEAR if (self.augment or r > 1) else cv2.INTER_AREA
683
  im = cv2.resize(im, (int(w0 * r), int(h0 * r)), interpolation=interp)
684
  return im, (h0, w0), im.shape[:2] # im, hw_original, hw_resized
685
+ return self.ims[i], self.im_hw0[i], self.im_hw[i] # im, hw_original, hw_resized
 
686
 
687
  def cache_images_to_disk(self, i):
688
  # Saves an image as an *.npy file for faster loading