py-faster-rcnn详解（2）——pascal _voc.py接口说明

最新推荐文章于 2023-05-28 13:29:55 发布

zchang81

最新推荐文章于 2023-05-28 13:29:55 发布

阅读量1.7k

点赞数

CC 4.0 BY-SA版权

分类专栏： faster rcnn 文章标签： Caffe 深度学习 py-faster-rcnn

本文链接：https://blog.youkuaiyun.com/zchang81/article/details/72904234

imdb对象是一个pascol _voc的类实例，该类继承自imdb，用于数据交互。

初始化函数

在初始化自身的同时，先调用了父类的初始化方法，将imdb _name传入，例如(‘voc_2007_trainval’)

  class pascal _voc(imdb):
  def __init__(self, image_set, year, devkit_path=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)
      self._classes = ('__background__', # always index 0
                       'aeroplane', 'bicycle', 'bird', 'boat',
                       'bottle', 'bus', 'car', 'cat', 'chair',
                       'cow', 'diningtable', 'dog', 'horse',
                       'motorbike', 'person', 'pottedplant',
                       'sheep', 'sofa', 'train', 'tvmonitor')
      self._class_to_ind = dict(zip(self.classes, xrange(self.num_classes)))
      self._image_ext = '.jpg'
      self._image_index = self._load_image_set_index()
      # Default to roidb handler
      self._roidb_handler = self.selective_search_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}

      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)

 
 
  
  1
  
  2
  
  3
  
  4
  
  5
  
  6
  
  7
  
  8
  
  9
  
  10
  
  11
  
  12
  
  13
  
  14
  
  15
  
  16
  
  17