arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2104.14965cs.CV
收藏

用于目标检测的无监督数据增强

Unsupervised data augmentation for object detection

National University of Singapore(新加坡国立大学)

Yichen Zhang, Zeyang Song, Wenbo Li

摘要:

数据增强一直是克服小数据集过拟合问题的有效方法。已经有很多增强操作,如水平翻转,随机裁剪甚至混合。然而,与图像分类任务不同,我们不能简单地执行这些操作的目标检测任务,因为缺乏相应的生成图像的标记的边界框信息。为了应对这一挑战,我们提出了一个框架,利用生成对抗网络(GAN)来执行无监督的数据增强。具体来说,基于YOLOv4最近的最高性能,我们提出了一个两步流水线,使我们能够生成一个图像,其中对象位于某个位置。这样就可以实现生成带有限界框标签的图像。

英文摘要:

Data augmentation has always been an effective way to overcome overfitting issue when the dataset is small. There are already lots of augmentation operations such as horizontal flip, random crop or even Mixup. However, unlike image classification task, we cannot simply perform these operations for object detection task because of the lack of labeled bounding boxes information for corresponding generated images. To address this challenge, we propose a framework making use of Generative Adversarial Networks(GAN) to perform unsupervised data augmentation. To be specific, based on the recently supreme performance of YOLOv4, we propose a two-step pipeline that enables us to generate an image where the object lies in a certain position. In this way, we can accomplish the goal that generating an image with bounding box label.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑