arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2102.00221cs.CV
收藏

ObjectAug:面向语义图像分割的对象级数据增强

ObjectAug: Object-level Data Augmentation for Semantic Image Segmentation

Fudan University(复旦大学);Guangzhou University(广州大学);Victoria University(维多利亚大学)

Jiawei Zhang, Yanchun Zhang, Xiaowei Xu

摘要:

语义图像分割的目标是获得具有精确边界的对象标签,通常会遇到过拟合问题。最近,各种数据增强策略,如区域辍学和混合策略已被提出来解决这个问题。这些策略已被证明是有效的指导模型参加歧视性较低的部分。然而,目前的策略是在图像级别上操作的,并且对象和背景是耦合的。因此,由于固定的语义场景,边界没有得到很好的增强。在本文中,我们提出ObjectAug进行对象级增强的语义图像分割。ObjectAug首先使用语义标签将图像分解为单个对象和背景。接下来,利用常用的增强方法(例如,缩放、移位和旋转)。然后,利用图像修补进一步恢复由对象增强带来的黑色区域。最后,将增强对象和背景组装为增强图像。通过这种方式,可以在各种语义场景中充分探索边界。此外,ObjectAug可以支持类别感知增强,为每个类别中的对象提供各种可能性,并且可以轻松地与现有的图像级增强方法相结合,以进一步提高性能。在自然图像和医学图像数据集上进行了全面的实验。实验结果表明,我们的ObjectAug可以明显提高分割性能。

英文摘要:

Semantic image segmentation aims to obtain object labels with precise boundaries, which usually suffers from overfitting. Recently, various data augmentation strategies like regional dropout and mix strategies have been proposed to address the problem. These strategies have proved to be effective for guiding the model to attend on less discriminative parts. However, current strategies operate at the image level, and objects and the background are coupled. Thus, the boundaries are not well augmented due to the fixed semantic scenario. In this paper, we propose ObjectAug to perform object-level augmentation for semantic image segmentation. ObjectAug first decouples the image into individual objects and the background using the semantic labels. Next, each object is augmented individually with commonly used augmentation methods (e.g., scaling, shifting, and rotation). Then, the black area brought by object augmentation is further restored using image inpainting. Finally, the augmented objects and background are assembled as an augmented image. In this way, the boundaries can be fully explored in the various semantic scenarios. In addition, ObjectAug can support category-aware augmentation that gives various possibilities to objects in each category, and can be easily combined with existing image-level augmentation methods to further boost performance. Comprehensive experiments are conducted on both natural image and medical image datasets. Experiment results demonstrate that our ObjectAug can evidently improve segmentation performance.

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

↑