Open HuShaohanAI opened 4 years ago
In pascal_voc.py `class pascal_voc(imdb): def init(self, image_set, year, devkitpath=None): imdb.init(self, 'voc' + year + '_' + image_set) self._year = year self._image_set = image_set self._devkit_path = self._get_default_path() if devkit_path is None \ else devkit_path self._data_path = os.path.join(self._devkit_path, 'VOC' + self._year)
if cfg.TRAIN.META_TYPE == 1: self._classes = ['__background__'] + cfg.TRAIN.ALLCLASSES_FIRST #second split if cfg.TRAIN.META_TYPE == 2: self._classes = ['__background__'] + cfg.TRAIN.ALLCLASSES_SECOND #third split if cfg.TRAIN.META_TYPE == 3: self._classes = ['__background__'] + cfg.TRAIN.ALLCLASSES_THIRD`
you always use ALLCLASSES, and dosenot filter class in phase 1, so you use base+novel class for training RPN of FasterRCNN?
In pascal_voc.py `class pascal_voc(imdb): def init(self, image_set, year, devkitpath=None): imdb.init(self, 'voc' + year + '_' + image_set) self._year = year self._image_set = image_set self._devkit_path = self._get_default_path() if devkit_path is None \ else devkit_path self._data_path = os.path.join(self._devkit_path, 'VOC' + self._year)
first split
you always use ALLCLASSES, and dosenot filter class in phase 1, so you use base+novel class for training RPN of FasterRCNN?