基于形状矫正和特征选择交叉注意力的端到端目标检测方法

本发明公开了基于形状矫正和特征选择交叉注意力的端到端目标检测方法，属于图像识别方法技术领域，该基于形状矫正和特征选择交叉注意力的端到端目标检测方法，包括步骤：主干网络提取输入图像特征；编码器对提取的图像特征进一步增强；解码器使用一组可学习的查询推理目标和图像全局上下文的关系，对查询进行迭代优化；检测头输出预测结果。本发明在模型中增加特征选择层，使用来自上一层解码器生成的预测框筛选图像特征，筛选出的特征用于计算下一层解码器交叉注意力，从而更准确地优化查询；引入形状矫正方法作为监督，将其作为一项新的损失加入总损失，使模型关注预测框与真实框的形状偏差，从而更准确地估计边界框。