一种基于YOLO算法的轻量化检测方法

本发明公开一种基于YOLO算法的轻量化检测方法，涉及计算机视觉与目标检测技术领域；包括：步骤1：对YOLOv5模型进行改进：将YOLOv5模型主干网络结构替换为MobileNet V3Small主干网络结构，将YOLOv5模型的Neck部分的Conv模块替换为轻量化卷积模块GSConv，在YOLOv5模型的C3部分增加NAMAttention注意力模块，步骤2：对改进后YOLOv5模型进行训练及优化，步骤3：使用优化后YOLOv5模型执行图像检测任务，输出检测结果。