基于双向特征融合金字塔SSD的铁矿石异物识别方法

    公开(公告)号:CN115601321A

    公开(公告)日:2023-01-13

    申请号:CN202211238672.7

    申请日:2022-10-11

    摘要: 本发明属图像识别技术领域,公开了基于双向特征融合金字塔SSD的铁矿石异物识别方法,包括将数据集划分为训练集和验证集;搭建主干网络,在SSD算法的基础上构建双向金字塔结构,提高网络的特征提取能力;融入轻量级坐标注意力模块,在卷积的过程中抓取关键特征,提升网络中特征的表征能力;优化Huber损失函数,提升网络的收敛速度、检测精度;得到基于双向特征融合金字塔的SSD算法;将数据集输入基于双向特征融合金字塔的SSD算法中,通过不断的训练学习,得到训练好的权重文件,实现对目标的识别。本发明实现了对高低层特征的双向融合,增强了网络的表征能力,提高了SSD网络的特征提取能力;提升网络的收敛速度和检测精度。