基于YOLOv5改进的主作业面人员检测方法
摘要:
本发明公开了一种基于YOLOv5改进的主作业面人员检测方法,包括:基于YOLOv5神经网络构建主作业面人员检测模型,其中,检测模型包括Input模块、Backbone网络模块、Neck网络模块和Output模块;采集主作业面场景的人员图像;将人员图像输入Input模块进行预处理;预处理后的人员图像通过Backbone网络模块进行切片处理,获取特征图;基于特征图,通过Neck网络模块进行采样融合,获取待预测特征图;将待预测特征图输入Output模块进行输出,获取人员图像的检测信息。本发明可以有效解决主作业面拍摄图像中小尺寸人员的检测问题,并提高检测的准确性和鲁棒性。
0/0