摘要
本发明提供的一种带语义引导的多尺度目标检测方法,包括:构建多尺度目标检测模型,多尺度检测模型包括主干网络、特征金字塔、检测头和语义引导分支;数据预处理;进行前向计算,得到多尺度目标检测模型的输出;将输出与真实目标进行损失函数的计算;使用反向传播算法计算所有可训练参数的梯度;使用梯度下降法更新模型的可训练参数,重复以上步骤,达到预设的次数结束训练,保存训练完成的模型;将待检测的图片输入训练好的模型,得到各分支的输出;对得到的各输出进行后处理,得到最终的检测结果。特征融合模块的检测头中设置两个特征选择模块,分别去学习当前层特征和下一层特征,从而减缓了不同尺度信息的梯度不一致性导致对训练时的干扰。
- 单位