From 03bec2b4260494178039b073de231e22bc9b86a3 Mon Sep 17 00:00:00 2001 From: ManoleAlexandru99 Date: Sat, 25 Mar 2023 20:38:55 +0200 Subject: [PATCH] Image Caching Disable #3 --- utils/dataloaders.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/dataloaders.py b/utils/dataloaders.py index ae7495562475..e631d3eb7fa5 100644 --- a/utils/dataloaders.py +++ b/utils/dataloaders.py @@ -599,7 +599,7 @@ def check_cache_ram(self, safety_margin=0.1, prefix=''): for _ in range(n): im = cv2.imread(random.choice(self.im_files)) # sample image ratio = self.img_size / max(im.shape[0], im.shape[1]) # max(h, w) # ratio - b += (im.nbytes * ratio ** 2) * 2 # ADDED *2 for mask - DELETE THIS IN CASE OF EMERGENCY + b += im.nbytes * ratio ** 2 mem_required = b * self.n / n # GB required to cache dataset into RAM mem = psutil.virtual_memory() cache = mem_required * (1 + safety_margin) < mem.available # to cache or not to cache, that is the question