扩展YOLOv5安全帽多级目标分类检测

作     者：金源 张长鲁 JIN Yuan;ZHANG Chang-Lu

作者机构：北京信息科技大学计算机学院北京100101 北京信息科技大学经济管理学院北京100192 

基　　金：国家重点研发计划(2018YFB1003705) 

出 版 物：《计算机系统应用》 (Computer Systems & Applications)

年 卷 期：2023年第32卷第2期

页      码：139-149页

摘      要：YOLO是目前计算机视觉目标检测领域比较重要的算法模型之一.基于现有YOLOv5s模型提出了一种扩展的YOLOv5多级分类目标检测算法模型.首先,对LabelImg标注工具进行功能扩展,使其满足多级分类标签文件构建;其次在YOLOv5s算法基础上修改检测头输出格式,在骨干网络前端引入DenseBlock、Res2Net网络模型核心设计思想,获取丰富的多维度特征信息,增强特征信息的重用性,实现了YOLO多级分类目标检测任务.在开源安全帽数据集上同时以安全帽颜色作为二级分类进行训练验证,平均精度,精确率和召回率分别达到了95.81%、94.90%和92.54%,实验结果验证了YOLOv5多级分类目标检测任务的可行性,并为目标检测及多级分类目标检测任务提供一种新的思路和方法.

主 题 词：目标检测 YOLOv5 多级分类 DenseBlock模块 损失函数 深度学习 

学科分类：12[管理学] 1201[管理学-管理科学与工程类] 08[工学] 081104[081104] 080203[080203] 0802[工学-机械学] 0835[0835] 0811[工学-水利类] 0812[工学-测绘类] 

D　O　I：10.15888/j.cnki.csa.008946

馆 藏 号：203118195...