科研之友
微信
新浪微博
Facebook
分享链接
摘要
本发明提出了一种基于曲率注意力机制的小目标检测方法,实现步骤为：获取训练样本集和测试样本集；构建基于曲率注意力机制的小目标检测网络模型；初始化参数；对目标检测网络模型进行训练；对目标检测网络模型的参数进行更新；获取训练好的目标检测网络模型和小目标的检测结果。本发明主干网络可以提取各种尺度的小目标特征,能够减少特征的丢失,曲率网络获取的每个样本的注意力图,可以补充分割图中模糊的边缘预测结果,注意力网络将主干网络获取的每个分割图与对应的曲率网络获取每个注意力图进行特征融合,避免了现有技术网络对关键边缘特征关注不足以及对目标信息提取不全面的缺陷,验结果表明,本发明能够有效提高小目标检测的准确率。
- 单位
