基于注意力机制的卷积神经网络胸部X光片异物检测方法

    公开(公告)号:CN115272215A

    公开(公告)日:2022-11-01

    申请号:CN202210863784.5

    申请日:2022-07-21

    Inventor: 杨柏林 罗文蔚

    Abstract: 本发明公开了基于注意力机制的卷积神经网络胸部X光片异物检测方法。本发明首先使用限制对比度自适应直方图均衡化算法对原图进行预处理;其次构建基于通道注意力的卷积神经网络,其中骨干网络进行特征提取生成多尺度特征图,特征金字塔结构进行多尺度特征融合,注意力模块在通道维度上,对多尺度特征图的特征融合进行加强,并将特征金字塔结构输出的特征图引入异物检测的头部;然后通过训练集数据对基于通道注意力的神经网络进行训练,得到训练好的网络模型;最后将测试集数据输入到训练好的网络模型,回归得到异物的坐标。本发明在通道维度上进行多尺度特征融合,充分利用每个通道的有效信息,准确对异物进行分类,并定位出异物的位置。

Patent Agency Ranking