ed1323aa16f4a5a1b17d32080ae40e1103ec6cca,gluon/coco_seg_dataset.py,COCOSegDataset,__init__,#COCOSegDataset#Any#Any#Any#,24

Before Change

        self.coco = COCO(annotations_file_path)
        self.coco_mask = mask
        if os.path.exists(idx_file_path):
            with open(idx_file_path, "rb") as f:
                self.ids = pickle.load(f)
        else:
            ids = list(self.coco.imgs.keys())
            self.ids = self._preprocess(ids, idx_file_path)
        self.transform = transform

After Change

        self.coco = COCO(annotations_file_path)
        self.coco_mask = coco_mask
        if os.path.exists(idx_file_path):
            self.idx = np.load(idx_file_path)
        else:
            idx_list = list(self.coco.imgs.keys())
            self.idx = self._filter_idx(idx_list, idx_file_path)

Instances