基于改进YOLOv5s轻量化模型的密集行人检测方法

本发明属于图像行人检测技术领域，具体涉及一种基于改进YOLOv5s轻量化模型的密集行人检测方法；包括：获取行人图像，将行人图像输入到改进YOLOv5模型中，得到初始密集行人检测结果；根据初始密集行人检测结果对改进YOLOv5模型进行结构化剪枝，得到中间改进YOLOv5模型；根据初始密集行人检测结果计算模型总损失，根据模型总损失调整中间改进YOLOv5模型参数，得到训练好的密集行人检测模型；采用Soft‑NMS算法对训练好的密集行人检测模型的检测结果进行筛选，得到最终的密集行人检测结果；本发明可以提高识别行人的准确率，有效的缓解行人之间出现的遮挡问题，可以在资源受限的设备上运行并且有着不错的效果。