def __init__()

in src/data_loader.py [0:0]

• 32 lines of code
• 4 McCabe index (conditional complexity)

    def __init__(self,
                 root,
                 annFile,
                 transform=None,
                 shuffle=False,
                 perm=None,
                 include_eos=False,):
        from pycocotools.coco import COCO
        self.root = root
        self.coco = COCO(annFile)
        self.ids = list(self.coco.imgs.keys())

        if perm is not None:
            self.ids = np.array(self.ids)[perm].tolist()

        self.transform = transform
        self.shuffle = shuffle

        self.category_list = [
            'eos', 'person', 'bicycle', 'car', 'motorcycle', 'airplane', 'bus', 'train', 'truck',
            'boat', 'traffic light', 'fire hydrant', 'stop sign', 'parking meter', 'bench', 'bird',
            'cat', 'dog', 'horse', 'sheep', 'cow', 'elephant', 'bear', 'zebra', 'giraffe',
            'backpack', 'umbrella', 'handbag', 'tie', 'suitcase', 'frisbee', 'skis', 'snowboard',
            'sports ball', 'kite', 'baseball bat', 'baseball glove', 'skateboard', 'surfboard',
            'tennis racket', 'bottle', 'wine glass', 'cup', 'fork', 'knife', 'spoon', 'bowl',
            'banana', 'apple', 'sandwich', 'orange', 'broccoli', 'carrot', 'hot dog', 'pizza',
            'donut', 'cake', 'chair', 'couch', 'potted plant', 'bed', 'dining table', 'toilet',
            'tv', 'laptop', 'mouse', 'remote', 'keyboard', 'cell phone', 'microwave', 'oven',
            'toaster', 'sink', 'refrigerator', 'book', 'clock', 'vase', 'scissors', 'teddy bear',
            'hair drier', 'toothbrush'
        ]

        self.category_list.append('<pad>')

        self.include_eos = include_eos

        # remove eos from category list if not needed
        if not self.include_eos:
            self.category_list = self.category_list[1:]