计算机工程与应用 ›› 2021, Vol. 57 ›› Issue (14): 176-180.DOI: 10.3778/j.issn.1002-8331.2005-0050

• 模式识别与人工智能 • 上一篇    下一篇

融合胶囊网络的文本-图像生成对抗模型

黄晓琪,王莉,李钢   

  1. 1.太原理工大学 大数据学院,山西 晋中 030600
    2.太原理工大学 软件学院,山西 晋中 030600
  • 出版日期:2021-07-15 发布日期:2021-07-14

Text-Image Generative Adversarial Model for Fusion Capsule Networks

HUANG Xiaoqi, WANG Li, LI Gang   

  1. 1.College of Data Science, Taiyuan University of Technology, Jinzhong, Shanxi 030600, China
    2.College of Software, Taiyuan University of Technology, Jinzhong, Shanxi 030600, China
  • Online:2021-07-15 Published:2021-07-14

摘要:

在传统文本-图像对抗模型的实现中,判别器中的卷积网络用于提取图像特征,但是卷积网络无法考虑到底层对象之间的空间关系,导致生成图像的质量较差,而胶囊网络是一种有效的解决方法。基于胶囊网络的方法对传统的文本条件式生成对抗网络模型进行了改进,将判别器中卷积网络换为胶囊网络,增强其对图像的鲁棒性。在Oxford-102和CUB数据集上的实验结果表明新模型可以有效提高生成质量,生成花卉图像的FID的数值降低了14.49%,生成鸟类的图像的FID的数值降低了9.64%。在Oxford-102和CUB两个数据集上生成图像的Inception Score分别提高了22.60%和26.28%,说明改进后模型生成的图片特征更丰富、更有意义。

关键词: 生成图像, 胶囊网络, 生成对抗网络, 卷积网络, 鲁棒性

Abstract:

In the implementation of the traditional text image confrontation model, the convolution network in the discriminator is used to extract image features, but the convolution network can not consider the spatial relationship between the underlying objects, resulting in poor quality of the generated image, and the capsule network is an effective method. In this paper, the traditional text conditional generation adversarial model is improved based on the capsule network method. The convolution network in the discriminator is replaced by the capsule network to enhance its robustness to image size. The experimental results on data set show that the new model can effectively improve the quality of generation. The FID value of flower image generation is reduced by 14.49%, and the FID value of bird image generation is reduced by 7.18%. The inception scores of the images generated on Oxford-102 and CUB data sets are 22.60% and 26.28% higher, which shows that the image features generated by the improved model are more rich and meaningful.

Key words: generating images, capsule network, generation adversarial network, convolutional network, robustness