Method __init__

eval/CIPO_evaluation/adapter.py:48–91

Constructor of Microsoft COCO helper class for reading and visualizing annotations. :param txtfile (str): location of annotation file :return:

(self, txtfile=None)

class adapter:
    def __init__(self, txtfile=None):
        """
        Constructor of Microsoft COCO helper class for reading and visualizing annotations.
        :param txtfile (str): location of annotation file
        :return:
        """
        # load dataset
        self.dataset,self.anns,self.cats,self.imgs = dict(),dict(),dict(),dict()
        self.imgToAnns, self.catToImgs = defaultdict(list), defaultdict(list)
        
        # anns : 
        #   annid -- imgid + idx int
        #       imgid
        #       bbox
        #       levelid
        #       annid 
        #       iscrowd 0

        # cats :
        #   id
        #   type    

        # imgs :
        #   file_path 
        #   imgid   int

        if not txtfile == None:
            print('loading annotations into memory...')
            tic = time.time()
            nms = []
            with open(txtfile, 'r') as f:
                line = f.readline().strip('\n')
                # print("Hello2")
                while line:
                    nms.append(line)
                    line = f.readline().strip('\n')
            # print(nms)

            self.createIndex(nms)
            # dataset = json.load(open(txtfile, 'r'))
            # assert type(dataset)==dict, 'annotation file format {} not supported'.format(type(dataset))
            # print('Done (t={:0.2f}s)'.format(time.time()- tic))
            # self.dataset = dataset
            # self.createIndex()
    def createIndex(self, files):
        print('creating index...')
        dataset, anns, cats, imgs = {}, {}, {}, {}