一种基于YOLOv5改进的无人机检测方法

本申请涉及一种基于YOLOv5改进的无人机检测方法、系统、设备及存储介质，涉及无人机检测领域。方法基于ImageNet数据集，其中方法包括：获取在包含遮挡物的背景下拍摄的无人机红外图像数据集；预处理并标注无人机红外图像数据集得到无人机红外图像增强数据集，无人机红外图像增强数据集包括训练集；构建基于YOLOv5的无人机检测模型；利用全卷积掩码自编码器训练无人机检测模型得到预训练模型；根据预训练模型和训练集逐层贪婪预训练并微调无人机检测模型得到无人机检测改进模型；利用无人机检测改进模型检测无人机红外图像数据集，得到检测结果。本申请的技术效果是：为了提高无人机红外图像检测的速度和精确度。