计算机与现代化Issue(6)：14-18,32,6.DOI:10.3969/j.issn.1006-2475.2024.06.003
基于半监督学习的StyleGAN图像生成模型
Semi-supervised Image Generation Model Based on StyleGAN
王志强 1郑爽1
作者信息
- 1. 北京工业大学信息学部,北京 100020
- 折叠
摘要
Abstract
This paper introduces SG-GAN,a semi-supervised StyleGAN model that overcomes the limitations of traditional StyleGAN.The quality of generated images using StyleGAN is heavily dependent on the quality of the training data set.When the training image quality is low,StyleGAN often fails to generate high-quality images.To address this issue,SG-GAN generates and trains support vector machine(SVM)training samples based on the one-to-one correspondence between vectors w and im-ages in StyleGAN.SVM and StyleGAN mapping network are then used to screen vectors w before generating each image to im-prove the quality of the resulting images.For batch image generation,gene vectors are generated by the gene vector generator and combined randomly while all permutations of style vectors are obtained using a dynamic cycle backtracking algorithm.Individuals are generated from the permutation results and screened for excellence using an evaluation function after multiple iterations.Ex-periments were carried out on open data sets and compared with other advanced methods,demonstrating that SG-GAN improves upon StyleGAN's accuracy significantly.The model achieves FID 2.74,an accuracy rate of 74.2%,and a recall rate of 51.2%on the lsun cat face data set,further validating the efficacy of the model.At the same time,the model achieved an accuracy of over 70%on the Cat Dataset,CIFAR-100,and ImageNet datasets,thereby verifying its good generalization ability.关键词
生成对抗网络/遗传算法/风格向量/支持向量机/动态循环回溯Key words
generative adversarial network/genetic algorithm/style vector/support vector machines/dynamic loop backtracking分类
信息技术与安全科学引用本文复制引用
王志强,郑爽..基于半监督学习的StyleGAN图像生成模型[J].计算机与现代化,2024,(6):14-18,32,6.
