Abstract
Due to the issue of UAV perspective, aerial traffic images often fail to achieve full-area full-angle coverage. In order to solve the problem of scarce and low-quality samples of aerial traffic images, data augmentation using image generation models has become a popular method for integrating advanced information technology in the transportation field. Currently, images generated by image generation models suffer from issues such as low image quality and difficulty in generating diverse samples. Therefore, to address these challenges, this paper proposes a new image generation model: the Autoencoder-Optimal transport model. This paper explains the image generation task from a geometric perspective, which involves two steps: manifold learning and probability distribution transformation. Firstly, an autoencoder is constructed to learn the underlying manifold. Secondly, a semi-discrete optimal transport network is established for probability distribution transformation. Finally, these two parts are combined to form an Autoencoder-Optimal transport model. Experimental results using aerial traffic images are analyzed to demonstrate the model's ability to generate realistic aerial traffic images.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
Similar content being viewed by others
References
Goodfellow, I., Pouget-Abadie, J., Mirza, M., et al.: Generative adversarial networks. Commun. ACM 63(11), 139–144 (2020)
Arjovsky, M., Chintala, S., Bottou, L.: Wasserstein generative adversarial networks. In: International Conference on Machine Learning, pp. 214–223. PMLR, Sydney (2017)
Rao, C.R.: Information and the accuracy attainable in the estimation of statistical parameters. In: Breakthroughs in Statistics, pp. 235–247. Springer, New York (1992)
Seung, H.S., Lee, D.D.: The manifold ways of perception. Science 290(5500), 2268–2269 (2000)
Lei, N., An, D., Guo, Y., Su, K., Liu, S., Luo, Z., Yau, S.T., Gu, X.: A geometric understanding of deep learning. Engineering 6(3), 361–374 (2020)
Pinaya, W.H.L., Vieira, S., Garcia-Dias, R., et al.: Autoencoders. In: Machine Learning, pp. 193–208. Academic Press, America (2020)
Kingma, D.P., Welling, M.: An introduction to variational autoencoders. Found. Trends® Mach. Learn. 12(4), 307–392 (2019)
Anwaar, M., Loo, C.K., Seera, M.: Face image synthesis with weight and age progression using conditional adversarial autoencoder. Neural Comput. Appl. 32, 3567–3579 (2020)
Dewi, C., Chen, R.C., Liu, Y.T., et al.: Synthetic Data generation using DCGAN for improved traffic sign recognition. Neural Comput. Appl. 34(24), 21465–21480 (2022)
Fetty, L., Bylund, M., Kuess, P., et al.: Latent space manipulation for high-resolution medical image synthesis via the StyleGAN. Z. Med. Phys. 30(4), 305–314 (2020)
Monge, G.: Géométrie descriptive, 2nd edn. J. Klostermann fils, Paris (1811)
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2024 Beijing Paike Culture Commu. Co., Ltd.
About this paper
Cite this paper
Zhang, Z., Jia, L., Qin, Y., Fan, X., Tang, T., Wang, Z. (2024). Data Augmentation of Aerial Traffic Images Based on Optimal Transport Theory. In: Gong, M., Jia, L., Qin, Y., Yang, J., Liu, Z., An, M. (eds) Proceedings of the 6th International Conference on Electrical Engineering and Information Technologies for Rail Transportation (EITRT) 2023. EITRT 2023. Lecture Notes in Electrical Engineering, vol 1138. Springer, Singapore. https://doi.org/10.1007/978-981-99-9319-2_41
Download citation
DOI: https://doi.org/10.1007/978-981-99-9319-2_41
Published:
Publisher Name: Springer, Singapore
Print ISBN: 978-981-99-9318-5
Online ISBN: 978-981-99-9319-2
eBook Packages: EngineeringEngineering (R0)