模型训练方法、红外弱小目标检测方法、装置及电子设备

本申请公开了一种模型训练方法及红外弱小目标检测方法，所训练的模型以Yolov3模型为框架，把Yolov3模型中对应大尺寸特征图的检测层剪除，使得模型更聚焦图像的中小目标；本申请还利用通道注意力机制和空间注意力机制，把在Yolov3模型的骨干网络提取的浅层特征视作有效特征，利用浅层特征指导中间层特征，使模型选择性关注在检测弱小目标时更重要的特征，并嵌入其他层有效特征，有效利用图像的细节信息，增强模型的拟合能力和检测性能；此外本发明还提出建立联合损失函数，缓解弱小目标位置回归约在相似度度量上的尺度敏感性，多个度量方式互相弥补，从多角度对位置预测框进行指导，有利于优化模型对目标最终位置预测的准确性。