From 7c92d1e8ed50989dc79826ddc3e3c7990176c4ee Mon Sep 17 00:00:00 2001 From: ManoleAlexandru99 Date: Sat, 25 Mar 2023 20:28:20 +0200 Subject: [PATCH] Image Caching Disable #2 --- utils/dataloaders.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/utils/dataloaders.py b/utils/dataloaders.py index 7dc254f9b305..ae7495562475 100644 --- a/utils/dataloaders.py +++ b/utils/dataloaders.py @@ -745,7 +745,7 @@ def __getitem__(self, index): def load_image(self, i): # Loads 1 image from dataset index 'i', returns (im, original hw, resized hw) - im, f, fn, seg = self.ims[i], self.im_files[i], self.npy_files[i], self.segs[i] + im, f, fn, seg = self.ims[i], self.im_files[i], self.npy_files[i], None if im is None: # not cached in RAM if fn.exists(): # load npy im = np.load(fn) @@ -767,7 +767,7 @@ def load_image(self, i): im = cv2.resize(im, (math.ceil(w0 * r), math.ceil(h0 * r)), interpolation=interp) seg = cv2.resize(seg, (math.ceil(w0 * r), math.ceil(h0 * r)), interpolation=interp) return im, (h0, w0), im.shape[:2], seg # im, hw_original, hw_resized - return self.ims[i], self.im_hw0[i], self.im_hw[i], self.segs[i] # im, hw_original, hw_resized + return self.ims[i], self.im_hw0[i], self.im_hw[i], seg # im, hw_original, hw_resized def cache_images_to_disk(self, i): # Saves an image as an *.npy file for faster loading