Detection Sample Augmentation Based on Building Index and Adversarial Network

Wei WANG , Dong-hua LU , Yan GAO , Yi-ting ZHANG

Science Technology and Engineering ›› 2024, Vol. 24 ›› Issue (3) : 1154 -1160.

PDF
Science Technology and Engineering ›› 2024, Vol. 24 ›› Issue (3) : 1154-1160 DOI: 10.12404/j.issn.1671-1815.2300102
research-article
Detection Sample Augmentation Based on Building Index and Adversarial Network
Author information +
History +
PDF

Abstract

In the large-scale target detection of remote sensing images based on deep learning, it is difficult to obtain some ground objects, and show poor performance in training results. Therefore, the morphological building index and the generative adversarial network were used for sample augmentation to reduce the problem of model overfitting caused by insufficient detection samples. By extracting the morphological building index related to the texture structure information, superimposing it with the original sample, the texture and spatial characteristics of the building could be strengthened. The existing samples were used to train the generative adversarial network to augment some targets categories. After compositing them with the samples enhanced by the morphological building index, the original sample set was expanded. Compared with the augmentation strategies of flipping, cropping, and changing the color, the detection accuracy of this method got 2% ~ 5% improvement on YLOLOv5, EfficientDet and other models. Experiments have proved that the sample augmentation method combining building index and generative adversarial network can significantly improve the detection accuracy of small-sample remote sensing image targets of special interest categories such as power stations.

Cite this article

Download citation ▾
Wei WANG, Dong-hua LU, Yan GAO, Yi-ting ZHANG. Detection Sample Augmentation Based on Building Index and Adversarial Network. Science Technology and Engineering, 2024, 24(3): 1154-1160 DOI:10.12404/j.issn.1671-1815.2300102

登录浏览全文

4963

注册一个新账户 忘记密码

References

Funding

PDF (4237KB)

15

Accesses

0

Citation

Detail

Sections
Recommended

/

〈 〉