Select-Mosaic: Data Augmentation Method for Dense Small Object Scenes

CV ODBO PDL
2024年06月08日
数据增强是指对原始数据应用一系列转换或扩展,以生成新的样本,从而增加数据的多样性和数量,有效提高模型的性能和鲁棒性。作为常见的数据增强方法,马赛克数据增强技术将多个图像拼接在一起,增加训练数据的多样性和复杂性,从而降低过拟合的风险。尽管马赛克数据增强通过拼接图像在一般检测任务中取得了出色的结果,但对于特定的检测任务仍存在一定的局限性。本文提出了Select-Mosaic数据增强方法来解决在航空图像中检测大量密集分布的小物体的挑战,该方法通过改进细粒度区域选择策略进行改进。改进后的Select-Mosaic方法在处理密集小物体检测任务方面表现出优异的性能,显著提高了检测模型的准确性和稳定性。代码可在https://github.com/malagoutou/Select-Mosaic获取。
Data augmentation refers to the process of applying a series of transformations or expansions to original data to generate new samples, thereby increasing the diversity and quantity of the data, effectively improving the performance and robustness of models. As a common data augmentation method, Mosaic data augmentation technique stitches multiple images together to increase the diversity and complexity of training data, thereby reducing the risk of overfitting. Although Mosaic data augmentation achieves excellent results in general detection tasks by stitching images together, it still has certain limitations for specific detection tasks. This paper addresses the challenge of detecting a large number of densely distributed small objects in aerial images by proposing the Select-Mosaic data augmentation method, which is improved with a fine-grained region selection strategy. The improved Select-Mosaic method demonstrates superior performance in handling dense small object detection tasks, significantly enhancing the accuracy and stability of detection models. Code is available at https://github.com/malagoutou/Select-Mosaic.
许愿