内容特征一致性风格迁移网络

摘要

针对编解码风格迁移网络在内容特征传递和跨域特征融合过程中易产生误差的问题,提出一种内容特征一致性风格迁移网络.将源域和目标域图像输入编码模块分别提取内容和风格特征;双链路特征传递模块在内容编码和解码模块间构建同级特征映射,利用特征增强和前馈参考链路并行处理内容特征;内容和风格特征在解码模块中融合并输出风格迁移结果.在多模态无监督图像转换(MUNIT)网络的基础上进行仿真测试,该网络对于低光照环境中目标模糊和目标与背景反差小的情况迁移效果提升明显.在BDD100K数据集上的实验结果表明,与MUNIT网络相比,所提网络的FID和IS指标分别平均下降3.2%和提升8.6%,能够实现图像内容一致、风格精确的图像变换,适用于无人驾驶等应用场景.