ACapsGan: Generative Adversarial Network Based on Capsule Network and Attention Mechanism
Rubo Jin, Jianda Cheng, Shiqi Chen, Jie Deng, Wei Wang · 2024
Large-scale, diverse and high-quality data is the foundation and key to achieving good generalization in target detection and recognition for deep learning-based algorithms. Directly collecting synthetic aperture radar (SAR) image data faces the difficulty in acquisition and high costs. Traditional SAR image simulation methods are limited by geometric and electromagnetic computation errors in their modeling process, and the high computational burden as well. Generative adversarial networks (GANs) offer a new approach for SAR image generation, but they struggle to achieve satisfactory results in terms of image quality and diversity. In order to overcome this problem, we propose a new type of GAN to learn the spatial relationship of the targets more effectively. Taking the real SAR images as input, we extract the target information through the capsule network, perturb the extracted features and adopt the attention mechanism to improve the quality and diversity of the augmented data.