在审

基于预训练单编码器的图像转换方法、装置、设备及介质

黄志坚、欧阳攀、栾悉道
长沙学院

摘要

本申请涉及一种基于预训练单编码器的图像转换方法、装置、设备及介质。所述方法包括:从源域数据集中读取原始图像,将原始图像输入预训练单编码器,得到五个不同尺寸的特征图;将五个不同尺寸的特征图输入小网络中进行特征融合,得到五个融合后的特征;利用五个融合后的特征中的多个特征构建判别器损失函数对判别器进行训练,得到更新后的图像转换模型,根据原始图像和循环图像构建第一生成器损失函数,根据原始图像和重建图像构建第二生成器损失函数,利用第一生成器损失函数和第二生成器损失函数对生成器进行训练,根据得到的最终的图像转换模型对待转换的图像进行图像转换。采用本方法能够提高图像转换准确率。

暂无引用专利