kakou.py 文件源码

def __init__(self, image_set, devkit_path=None):
        datasets.imdb.__init__(self,image_set)#imageset ?train  test
        self._image_set = image_set
        self._devkit_path = devkit_path
        self._data_path = os.path.join(self._devkit_path)
        self._classes = ('__background__','car','person','bike', 'truck', 'van', 'tram', 'misc')#????
        self._class_to_ind = dict(zip(self.classes, xrange(self.num_classes)))#????{'__background__':'0','car':'1'}
        if self._image_set=='KakouTrain':
            self._image_index = self._load_image_set_index('KITTI_train_list.txt')#??????
        elif self._image_set=='KakouTest':
            self._image_index = self._load_image_set_index('KITTI_val_list.txt')#??????
        # Default to roidb handler
        self._roidb_handler = self.selective_search_roidb
        # PASCAL specific config options
        self.config = {'cleanup'  : True,
                   'use_salt' : True,
                   'top_k'    : 2000,
                   'use_diff' : False,
                   'rpn_file' : None}
        assert os.path.exists(self._devkit_path), \
            'VOCdevkit path does not exist: {}'.format(self._devkit_path)
        assert os.path.exists(self._data_path), \
            'Path does not exist: {}'.format(self._data_path)