7

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

我怎样才能最好地将我的增强序列与 api 集成以进行即时训练?

例如,假设我有一个增强器:

aug = iaa.SomeOf((0, 2), 
                 [iaa.Fliplr(0.5), iaa.Flipud(0.5), iaa.Affine(rotate=(-10, 10))])

有什么方法可以配置对象检测 api 来使用它吗?

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

4

1 回答 1

0

有人为此做了一个回购:
https ://github.com/JinLuckyboy/TensorFlowObjectDetectionAPI-with-imgaug

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

于 2021-01-18T20:39:38.043 回答