基于自适应标签分配的红外船舶检测方法、装置及设备

本发明涉及图像处理技术领域，公开了一种基于自适应标签分配的红外船舶检测方法、装置及设备。该方法包括：获取红外船舶图像数据集；构建初始船舶检测模型，并将红外船舶图像数据集输入初始船舶检测模型；采用高斯感受野标签分配策略对初始船舶检测模型进行训练并对损失函数进行改进，得到训练后的船舶检测模型；根据训练后的船舶检测模型对实时红外图像进行检测，以输出船舶检测结果。本发明中在模型训练阶段采用高斯感受野标签分配策略进行自适应分配标签，为不同大小的目标采取不同的标签分配形式，匹配更有效的正样本，在兼顾中大目标船舶检测的同时对小目标船舶检测进行优化，并改进损失函数，提升了模型对红外船舶的整体检测性能。