科研之友
微信
新浪微博
Facebook
分享链接
摘要
对于电力等工业场景中的巡检任务，需要将虚拟仿真生成的虚拟图像转换为真实的风格，辅助接下来的设备定位和设备缺陷检测工作。然而传统的图像风格迁移方法大多着眼于艺术图像与现实场景的转换，由于艺术图像本身的特性，在迁移前后图像的内容结构可能会出现一定程度的变化，这种变化会对下游任务的性能造成负面影响，因而无法直接应用。因此本文提出一种基于对比学习的图像风格迁移方法，通过在多个层次的特征上分别对比相同与不同位置图像块的特征，能够使生成图像在深层特征上向目标域靠近，同时约束迁移前后的图像内容上保持一致。实验表明，本文所提出的方法能够在保证图像主体内容不发生变形和缺失的同时，节省模型训练过程中的计算开销。
-
单位国网安徽省电力有限公司; 中国科学技术大学先进技术研究院
