一种基于改进YOLOv5s的轻量化行人检测方法及装置
申请号:CN202510890635
申请日期:2025-06-30
公开号:CN120783283A
公开日期:2025-10-14
类型:发明专利
摘要
本发明公开了一种基于改进YOLOv5s的轻量化行人检测方法及装置,所述方法包括以下步骤:(1)构建多场景行人图像数据集;(2)为数据集中的图像添加标注信息;(3)构建基于改进YOLOv5s的轻量化行人检测模型;(4)采用训练集和验证集对所述检测模型进行训练,并将训练好的模型参数保存为最优模型;(5)采用测试集对最优模型进行测试,测试结果满足精度要求,即获得最终的基于改进YOLOv5s的轻量化行人检测模型。相较于现有技术,本发明公开的一种基于改进YOLOv5s的轻量化行人检测方法,实现了模型的轻量化,降低了计算量,同时提高了模型检测精度。
技术关键词
行人检测模型
行人检测方法
Sigmoid函数
客观评价指标
注意力机制
通道
模型训练模块
采样模块
输出特征
融合空间信息
行人特征提取
数据获取模块
行人检测装置
图像
样本
特征融合网络
更新模型参数