一种基于YOLO的轻量级目标检测方法

本发明涉及计算机视觉技术领域，提供一种基于YOLO的轻量级目标检测方法。包括以下步骤：获取并预处理图像，获得预处理后的图像；基于YOLOv8框架建立轻量级目标检测模型，所述轻量级目标检测模型包括主干网络、特征提取网络和联合剪枝‑优化模块；设置训练参数，使用预处理后的图像对所述轻量级目标检测模型进行训练，得到轻量级目标检测训练模型；将需要的检测图像输入到轻量级目标检测训练模型中，得到检测结果。本发明通过替换骨干特征提取网络提高模型的检测精度和特征融合能力。然后利用模型稀疏化训练和模型剪枝，提升推理速度降低网络参数，通过知识蒸馏恢复模型精度并保持模型轻量性。