摘要
本说明书实施例公开了一种VAE的训练方法,该方法将样本图像输入待训练VAE模型,通过VAE模型的编码模块将该样本图像映射到隐空间,得到该样本图像对应的隐变量分布,再通过VAE模型的解码模块将隐变量分布转换为图像,以生成重构图像,从而可分别从重构图像和样本图像中提取第一图像特征和第二图像特征,根据两个图像特征之间的差异确定感知损失,并根据感知损失对待训练VAE模型进行训练。通过上述方法可准确的训练出高精度的VAE模型。
技术关键词
样本
解码模块
编码模块
变量
精度优化方法
生成重构图像
图像增强
特征提取模型
噪声
处理器
重构模块
图像压缩
颜色
编码器
训练装置
电子设备
可读存储介质