基于改进注意力W-Net的工业烟尘图像分割
来源：计算机集成制造系统杂志2023年第2期北京时间：
作者：詹光莉, 刘辉, 杨路
单位：昆明理工大学信息工程与自动化学院
摘要：针对小目标烟尘尺寸小、边缘稀薄和U-Net在提取小目标烟尘特征效果不佳等原因导致的烟尘漏检、误检和分割精度低等问题,提出一种基于改进注意力W-Net(IAW-Net)的烟尘图像分割网络。采用注意力机制将U-Net扩展为W-Net,在W-Net的基础上引入改进的注意力机制,增强了小目标烟尘的特征;针对小目标烟尘特点对焦点损失进行改进,增加了小目标烟尘的分割比重。实验结果表明,IAW-Net能够在不影响大目标烟尘分割的基础上更加关注小目标烟尘的分割效果,从而提升了烟尘图像的整体分割能力,相比现有语义分割网络具有更好的分割效果。
关键词：工业烟尘,图像分割,W-Net,注意力机制,焦点损失
基金资助：国家自然科学基金资助项目(61863018,62263016);云南省科技厅应用基础研究资助项目(202001AT070038)。
