行人检测模型训练方法、装置、设备及存储介质

AITNT
正文
推荐专利
行人检测模型训练方法、装置、设备及存储介质
申请号:CN202411899901
申请日期:2024-12-23
公开号:CN120088612A
公开日期:2025-06-03
类型:发明专利
摘要
本申请涉及一种行人检测模型训练方法、装置、设备及存储介质,应用在深度学习领域,包括获取行人数据集,并将所述行人数据集根据预设的比例划分为行人数据训练集、行人数据验证集和行人数据测试集;基于Yolov5模型构建待训练的行人检测模型,采用所述行人数据训练集和所述行人数据验证集对所述待训练的行人检测模型进行训练,并生成训练完成的行人检测模型;利用所述行人数据测试集对所述训练完成的行人检测模型进行测试,并获得所述训练完成的行人检测模型的测试结果。本申请具有的技术效果是:提高行人危险程度检测的准确率,提高模型的泛化性。
技术关键词
行人检测模型 数据验证 金字塔池化模块 生成图像特征 网络 训练集 视频 模型训练模块 数据获取模块 注意力机制 训练装置 测试模块 处理器 计算机设备 可读存储介质 存储器
添加客服微信openai178,进AITNT官方交流群
驱动智慧未来:提供一站式AI转型解决方案
沪ICP备2023015588号