基于风格迁移的交互式航空发动机孔探图像扩展方法

【类型】期刊

【作者】樊玮 段博坤 黄睿 刘挺 张宁  

【刊名】计算机应用

【关键词】 交互式,泊松融合,数据扩展,风格迁移,孔探图像

【摘要】在航空发动机孔探图像缺陷检测任务中,缺陷区域样本数量远少于正常样本数量,且缺陷样本无法覆盖整个样本空间,导致检测算法泛化能力较差。针对上述问题,提出了一种基于风格迁移技术的交互式数据扩展方法。首先,通过交互界面选择背景图像和缺陷目标,并根据背景图像指定需要粘贴的目标的大小、角度和位置等信息;其次,通过风格迁移技术将背景图像的风格迁移到目标图像上,使得背景图像和待检测目标具有相同的风格;最后,利用泊松融合算法对融合区域的边界进行修正,以达到连接区域自然过渡的效果。通过二分类和缺陷检测验证了该方法的有效性。在包含真实图像和扩展后图像的二分类实验中,测试人员的平均分类错误率达到了44.0%;而在基于Mask R-CNN模型的检测任务中,所提方法的检测和分割平均精度(AP)较传统方法分别提高了99.5%和91.9%。

【年份】2021

【作者单位】中国民航大学计算机科学与技术学院;厦门航空信息部;

【期号】12

【页码】3631-3636

【全文挂接】全文挂接