[发明专利]一种基于注意力机制的图像故障细粒度识别方法

摘要

本发明提供一种基于注意力机制的图像故障细粒度识别方法，在细粒度识别网络NTS‑Net的基础上，使用一种软池化的双通道注意力网络，针对铁路数据集黑、灰、白的背景，经过两个通道，从两个维度获取更加有效的特征语义。为了避免特征信息在经过注意力网络的时候有所丢失，通过软池化的方式来避免池化过程中的信息损失，有助于后续环节。最后，通过更改网络损失函数，在不需要对故障图片进行数据增强的前提下，削弱故障样本和正常样本数量不平衡给结果带来的干扰。该识别方法采用的是弱监督方法，高效快捷，适用范围广，算力消耗低。