bfbd2223521126cefe0a25382f09a684c34e4a01,mmdet/datasets/coco.py,CocoDataset,_filter_imgs,#CocoDataset#Any#,94

Before Change

        Filter images too small or without ground truths.
        img_ids = list(set([_["image_id"] for _ in self.coco.anns.values()]))
        valid_ids = []
        img_infos = []
        for i in img_ids:
            info = self.coco.loadImgs(i)[0]
            if min(info["width"], info["height"]) >= min_size:
                valid_ids.append(i)
                img_infos.append(info)
        return valid_ids, img_infos

    def _load_ann_info(self, idx):
        img_id = self.img_ids[idx]

After Change

        Filter images too small or without ground truths.
        valid_inds = []
        ids_with_ann = set(_["image_id"] for _ in self.coco.anns.values())
        for i, img_info in enumerate(self.img_infos):
            if self.img_ids[i] not in ids_with_ann:
                continue
            if min(img_info["width"], img_info["height"]) >= min_size:
                valid_inds.append(i)
        return valid_inds

    def _parse_ann_info(self, ann_info, with_mask=True):
        Parse bbox and mask annotation.

Instances