armpos.py 文件源码

def __init__(self, data_path=None):
        imdb.__init__(self, 'armpos')
        self._data_path = data_path if data_path is not None else os.path.join(cfg.DATA_DIR, 'armpos')
        self._classes = ('__background__', # always index 0
                            "armpos"
                             )
        self._class_to_ind = dict(zip(self.classes, xrange(self.num_classes)))
        self._image_index = self._load_image_set_index()
        self._image_ext = '.jpg'
        # Default to roidb handler
        #self._roidb_handler = self.selective_search_roidb
        self._roidb_handler = self.gt_roidb
        self._salt = str(uuid.uuid4())

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