Tensorflow对象检测API:如何使用Imgaug进行扩增?

eri*_*ric 5 tensorflow object-detection-api data-augmentation

我一直在使用imgaug手动滚动增强器,因为我真的很喜欢tf对象检测api中不可用的一些选项。例如,我使用运动模糊,因为我的很多数据都有快速移动的模糊对象。

如何最好地将扩充序列与api集成在一起以进行实时训练?

例如,说我有一个增强器:

aug = iaa.SomeOf((0, 2), 
                 [iaa.Fliplr(0.5), iaa.Flipud(0.5), iaa.Affine(rotate=(-10, 10))])
Run Code Online (Sandbox Code Playgroud)

有什么方法可以配置对象检测API来解决此问题?

我当前正在做的是imgaug用来生成(增强)训练数据,然后tfrecord从该扩充管道的每次迭代中创建文件。这是非常低效的,因为在培训期间,我将大量数据保存到磁盘上,而不是即时运行扩充。

eri*_*ric 0

有人为此创建了一个存储库:
https://github.com/JinLuckyboy/TensorFlowObjectDetectionAPI-with-imgaug

抱歉,这不是代码答案,我还没有真正研究过它,所以我不会将其标记为正式答案。如果我有机会测试它,我会让人们知道。