coco_dataset.py 文件源码

def _load_coco_annotation(self, index):
        annIds = self.COCO.getAnnIds(imgIds=index, iscrowd=None)
        objs = self.COCO.loadAnns(annIds)
        bag_labels = np.zeros(self.num_classes)
        for obj in objs:
            cls = self.coco_cat_id_to_class_ind[obj['category_id']]
            bag_labels[cls] = 1
        return bag_labels