一种基于红外图像的轻量化目标检测方法、装置及设备

本说明书公开了一种基于红外图像的轻量化目标检测方法、装置及设备，涉及计算机视觉与人工智能领域。该方法通过构建无人机红外图像目标检测数据集及图像预处理，将数据集划分为训练集、验证集和测试集；构建双流YOLOv5m网络和剪枝后的YOLOv5n网络；使用无人机训练集分别训练双流YOLOv5m网络和剪枝后YOLOv5n网络，得到初始教师模型和初始学生模型；利用初始教师模型对初始学生模型进行知识蒸馏训练得到优化后的学生模型，将优化后的学生模型应用于无人机测试集中。解决了现有红外目标检测方法模型较为臃肿，模型参数量大，不能满足边缘计算平台使用的问题。该方法有减少网络的参数数量和计算复杂度，在算力受限的情况下显著提升红外无人机图像目标检测的速度和准确率。