add load_cache functino in dataset

dataset/build.py

@@ -36,7 +36,8 @@ def build_dataset(args, data_cfg, trans_config, transform, is_train=False):
             data_dir=data_dir,
             image_sets=[('2007', 'trainval'), ('2012', 'trainval')] if is_train else [('2007', 'test')],
             transform=transform,
-            trans_config=trans_config
+            trans_config=trans_config,
+            load_cache=args.load_cache
             )
     ## COCO dataset
     elif args.dataset == 'coco':
@@ -45,7 +46,8 @@ def build_dataset(args, data_cfg, trans_config, transform, is_train=False):
             data_dir=data_dir,
             image_set='train2017' if is_train else 'val2017',
             transform=transform,
-            trans_config=trans_config
+            trans_config=trans_config,
+            load_cache=args.load_cache
             )
     ## Custom dataset
     elif args.dataset == 'ourdataset':
@@ -55,6 +57,7 @@ def build_dataset(args, data_cfg, trans_config, transform, is_train=False):
             image_set='train' if is_train else 'val',
             transform=transform,
             trans_config=trans_config,
+            load_cache=args.load_cache
             )
 
     return dataset, dataset_info

dataset/coco.py

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