Tensorflow对象检测API中的数据扩充

时间:2018-05-11 13:10:19

标签: python tensorflow object-detection object-detection-api

在配置文件中,我们将获得默认的扩充选项,如下所示。

data_augmentation_options {
    random_horizontal_flip {
    }
  }

但是我想知道它如何与训练图像给出的边界框(地面实况框)值一起工作。所以我看了preprocessor.py,random_horizo​​ntal_flip()接受'boxes = None'参数。由于在配置文件中没有给出参数,我认为当它执行随机水平翻转时,此翻转不会考虑边界框。

我的问题是我在上面显示的代码片段中使用什么参数在配置文件中添加边界框的值。

1 个答案:

答案 0 :(得分:0)

盒子也会翻转。如果向下看预处理器文件,您会注意到一个映射,它定义了张量字典的哪些输入将传递到预处理函数中。 groundtruth框传递给random_horizo​​ntal_flip。