摘要
本申请涉及一种行人检测模型训练方法、装置、设备及存储介质,应用在深度学习领域,包括获取行人数据集,并将所述行人数据集根据预设的比例划分为行人数据训练集、行人数据验证集和行人数据测试集;基于Yolov5模型构建待训练的行人检测模型,采用所述行人数据训练集和所述行人数据验证集对所述待训练的行人检测模型进行训练,并生成训练完成的行人检测模型;利用所述行人数据测试集对所述训练完成的行人检测模型进行测试,并获得所述训练完成的行人检测模型的测试结果。本申请具有的技术效果是:提高行人危险程度检测的准确率,提高模型的泛化性。
技术关键词
行人检测模型
数据验证
金字塔池化模块
生成图像特征
网络
训练集
视频
模型训练模块
数据获取模块
注意力机制
训练装置
测试模块
处理器
计算机设备
可读存储介质
存储器