Attentive Mix: An Efficient Data Augmentation Method for Object Detection
Runji Liu, Ying Chen, Jiasheng Wang +1 more
Data augmentation is an import method to improve the model performance, many different data augmentation algorithms have been proposed in the field of machine learning, such as Mixup, Cutmix, Cutout, Mosaic, Attentive Cutmix, Dropout, DropBlock. The above algorithms are mainly applied in image classification tasks, but there is still lack of a data augmentation algorithm that can be used directly for object detection tasks. We propose an efficient data augmentation algorithm specifically applied in the field of object detection. Specifically, we select two images from dataset randomly, then use a trained model to generate the attention map from an image and paste it into another image to generate a new image. Finally, we train a new model on the original data with new generated data. The advantage is that it does not lead to discontinuities in the bounding box when applied to object…