debug cache

dataset/coco.py

@@ -102,11 +102,11 @@ class COCODataset(Dataset):
         self.cached_images = []
         self.cached_targets = []
         dataset_size = len(self.ids)
+
         print('loading data into cache ...')
         for i in range(dataset_size):
-            if i % 5000 == 0 and i > 0:
+            if i % 5000 == 0:
                 print("[{} / {}]".format(i, dataset_size))
-                break
             # load an image
             image, image_id = self.pull_image(i)
             orig_h, orig_w, _ = image.shape

dataset/ourdataset.py

@@ -86,6 +86,7 @@ class OurDataset(Dataset):
         self.cached_targets = []
         dataset_size = len(self.ids)
 
+        print('loading data into memory ...')
         for i in range(dataset_size):
             if i % 5000 == 0:
                 print("[{} / {}]".format(i, dataset_size))

dataset/voc.py

@@ -144,6 +144,7 @@ class VOCDetection(data.Dataset):
         self.cached_targets = []
         dataset_size = len(self.ids)
 
+        print('loading data into memory ...')
         for i in range(dataset_size):
             if i % 5000 == 0:
                 print("[{} / {}]".format(i, dataset_size))