Skip to main content
 首页 » 编程设计

python之Tensorflow 对象检测 API 中的数据增强

2024年09月07日35cyq1162

在配置文件中,我们获得了默认的增强选项,如下所示。

data_augmentation_options { 
    random_horizontal_flip { 
    } 
  } 

但我想知道它如何与训练图像给出的边界框(地面真实框)值一起使用。所以我看了preprocessor.py , random_horizo​​ntal_flip() 采用 'boxes=None' 参数。由于配置文件中没有给出参数,我假设这个翻转在进行随机水平翻转时不考虑边界框。

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

请您参考如下方法:

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