yjh0410 пре 2 година
родитељ
комит
20fbe8401d
1 измењених фајлова са 1 додато и 0 уклоњено
  1. 1 0
      dataset/coco.py

+ 1 - 0
dataset/coco.py

@@ -72,6 +72,7 @@ class COCODataset(Dataset):
         self.ids = self.coco.getImgIds()
         self.class_ids = sorted(self.coco.getCatIds())
         self.is_train = is_train
+        self.load_cache = load_cache
 
         # augmentation
         self.transform = transform