Faster R-CNN中 pascal_voc.py 的代码详细解析

最新推荐文章于 2022-06-07 17:57:26 发布

原创

最新推荐文章于 2022-06-07 17:57:26 发布 · 2.9k 阅读

8 ·

CC 4.0 BY-SA版权

文章标签：

#Python

本文详细解析Faster R-CNN中pascal_voc.py类的16个函数，包括init()、image_path_at()、gt_roidb()等，涉及图像路径、数据集索引、标签读取、结果文件生成及AP计算等内容，旨在理解代码功能和流程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

从下图可以看出来，pascal_voc实际上是集成了imdb类的一个类。所谓继承，就是儿子可以用爸爸的身份证，注册各种游戏~也就是这个类可以里面不用定义一些def（一些功能代码段），这些def在其父亲类就定义了，在这个类里可以直接使用，去写更复杂的类，继承，说白了就是这个类调用了另一个类。

一共有16个函数定义，其中有一些是函数，有一些就是定义了一些类的参数。

——————————16个函数逐个解析————————

首先设置了编号，按编号来。

1.init() 生成类的一些参数，看到下代码，全是等于号，就是赋值。设置初始参数。就是在内存中先占这么个坑，以后要放东西的。

  def __init__(self, image_set, year, use_diff=False):
    name = 'voc_' + year + '_' + image_set
    if use_diff:
      name += '_diff'
    imdb.__init__(self, name)
    self._year = year
    self._image_set = image_set
    self._devkit_path = self._get_default_path()
    self._data_path = os.path.join(self._devkit_path, 'VOC' + self._year)
    self._classes = ('__background__',  # always index 0
                     'aeroplane', 'bicycle', 'bird', 'boat',