building.py 文件源码

def __init__(self, image_set, building_path=None):
        datasets.imdb.__init__(self, 'building_' + image_set)

        self._image_set = image_set     # something like 'train', 'val', ...

        self._building_path = self._get_default_path() if building_path is None \
                            else building_path
        # building_path = ... Faster-RCNN_TF/data/building_data


        self._classes = ('__background__', 'building')
        self._class_to_ind = dict(zip(self.classes, xrange(self.num_classes)))   # {'Building': 1, '__background__': 0}
        self._image_ext = '.png'
        self._image_index = self._load_image_set_index()    # something like: ['22', '23', '29']

        # Default to roidb handler
        """if cfg.IS_RPN:
            self._roidb_handler = self.gt_roidb
        else:
            self._roidb_handler = self.region_proposal_roidb"""

        self._roidb_handler = self.gt_roidb

        self._salt = str(uuid.uuid4())
        self._comp_id = 'comp4'

        # PASCAL specific config options
        self.config = {'cleanup': True,
                       'use_salt': True,
                       'use_diff': False,
                       'matlab_eval': False,
                       'rpn_file': None,
                       'min_size': 2}






    ####################################################################################################################
    # ---------------------------------------------------------------------------------------------------------------- #