glenn-jocher
commited on
Commit
•
0ab303b
1
Parent(s):
916bdb1
Remove `else:` from load_image() (#8692)
Browse files- 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 |
-
|
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
|