Application of SAR Ship Data Augmentation Based on Generative Adversarial Network in Improved SSD

被引：0
|
作者：
Yang L. [1 ]
Su J. [1 ]
Li X. [1 ]
机构：
[1] College of Nuclear Engineering, Rocket Force University of Engineering, Xi'an, 710025, Shaanxi
来源：
Binggong Xuebao/Acta Armamentarii | 2019年 / 40卷 / 12期
关键词：
Generative adversarial network; Object detection; Pix2pix; Ship; Single shot multibox detector; Synthetic aperture radar;
D O I：
10.3969/j.issn.1000-1093.2019.12.013
中图分类号：
学科分类号：
摘要：
For the high cost of data acquisition and few of datasets for ship detection in synthetic aperture radar (SAR) image, a data augmentation technology based on pix2pix is proposed. A dataset is set for pix2pix, and 800 SAR ship samples are obtained by training and testing the generative adversarial network (GAN). The objective evaluation is given for the generated typical samples. And for the problems that the accuracy of traditional ship detection in SAR images is susceptible to speckle noise and its generalization is poor, a ship detection algorithm based on single shot multibox detector (SSD) is proposed. An Inception module is added into the SSD detecting algorithm for enhancing its adaptability to multi-size target and improving the performance of detector. Finally, the SAR ship data generated by pix2pix GAN is marked and added to the improved SSD. A large number of comparison experiments were performed on the SSDD dataset. The experimental results show that the detection accuracy is improved by 4.3% when the generated samples are added to SSD. The detection accuracy is improved by 1.9% after the samples are added to the improved SSD; and the detection accuracy of the improved SSD is improved by 4.7% without the addition of the generated sample in the detector. © 2019, Editorial Board of Acta Armamentarii. All right reserved.
引用
收藏
页码：2488 / 2496
页数：8
相关论文
共 23 条
  • [1] Erhan D., Szegedy C., Toshev A., Et al., Scalable object detection using deep neural networks, Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 2147-2154, (2014)
  • [2] Li J.W., Qu C.W., Peng S.J., Et al., Ship detection in SAR images based on convolutional neural network, Systems Engineering and Electronics, 40, 9, pp. 1953-1959, (2018)
  • [3] Ren S., He K., Girshick R., Et al., Faster R-CNN: towards real-time object detection with region proposal networks, IEEE Transactions on Pattern Analysis & Machine Intelligence, 39, 6, pp. 1137-1149, (2015)
  • [4] Li J.W., Qu C.W., Peng S.J., Et al., Ship detection in SAR images based on generative adversarial network and online hard examples mining, Journal of Electronics and Information Technology, 41, 1, pp. 143-149, (2019)
  • [5] Goodfellow I., Pouget-Abadie J., Mirza M., Et al., Ge-nerative adversarial nets, Proceedings of Advances in Neural Information Processing Systems, pp. 2672-2680, (2014)
  • [6] Girshick R., Fast R-CNN, Proceedings of the IEEE International Conference on Computer Vision, pp. 1440-1448, (2015)
  • [7] Kang M., Ji K.F., Leng X.G., Et al., Contextual region-based convolutional neural network with multilayer fusion for SAR ship detection, Remote Sensing, 9, 8, pp. 1-14, (2017)
  • [8] Jiao J., Zhang Y., Sun H., Et al., A densely connected end-to-end neural network for multiscale and multiscene SAR ship detection, IEEE Access, 6, pp. 20881-20892, (2018)
  • [9] Radford A., Metz L., Chintala S., Unsupervised representation learning with deep convolutional generative adversarial networks, Proceedings of the 4th International Conference on Learning Representation, (2016)
  • [10] Chen X., Duan Y., Houthooft R., Et al., InfoGAN: interpretable representation learning by information maximizing generative adversarial nets, Proceedings of Advances in Neural Information Processing Systems, pp. 2172-2180, (2016)
123