计算机应用

作者：樊玮, 段博坤, 黄睿, 刘挺, 张宁

单位：1. 中国民航大学 计算机科学与技术学院, 天津 300300;2. 厦门航空 信息部, 福建 厦门 361006

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

基金：天津市教委科研计划项目（2019KJ126）；中国民航大学中央高校基金资助项目（3122018C020）；中国民航大学科研启动基金资助项目（600001050115）。

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

来源：2020年第12期

《计算机应用》期刊编辑部

查看计算机应用杂志2020年第12期