摘要
本发明公开了基于可变形注意力特征融合的小目标检测方法,步骤S100、获取数据集并进行数据预处理;步骤S200、构建基于可变形注意力机制的增强小目标肺结节检测网络模型;步骤S300、训练基于可变形注意力机制的增强小目标肺结节检测网络模型;步骤S400、将待检测图像输入训练好的基于可变形注意力机制的增强小目标肺结节检测网络模型,输出检测结果。本发明针对尺寸较小目标特征集中于浅层特征中的特点,在特征融合金字塔中增加了四倍下采样的检测层,补充原本检测层中遗漏的小目标的空间位置等信息,有助于提取和增强小目标的特征表示,同时减少背景冗余特征的干扰,提高小目标的检测准确性。
技术关键词
结节检测
注意力机制
网络
传播算法
采样模块
双线性插值
置信度阈值
冗余特征
像素
数据
图像
矩阵
金字塔
优化器
采样点
特征点
通道
尺寸