| 针对CenterNet缺点的安全帽检测算法改进 |
| |
| 引用本文： | 王海瑞 ?,赵江河,吴蕾,谢思远.针对CenterNet缺点的安全帽检测算法改进[J].湖南大学学报(自然科学版),2023(8):125-133. |
| |
| 作者姓名： | 王海瑞 ? 赵江河 吴蕾 谢思远 |
| |
| 作者单位： | （昆明理工大学 信息工程与自动化学院，云南 昆明 650504） |
| |
| 摘 要： | 为解决安全帽数据集上识别率较低的问题，本文提出了一种基于改进CenterNet网络结构的检测方法.针对CenterNet多分类时热度图预测结果较差的问题，本文尝试在损失函数上进行改进，提出了Focal-Mse-One损失以及Focal-Mse-Guss损失，并且对这两个损失函数以及原始的Focal损失函数进行对比；针对CenterNet对推理过程中特征图复用性不高的问题提出了ASFF和DASFF结构并进行比较.实验结果表明，在GeForce GTX 1050显卡上推理速度可以达到20.78帧.在安全帽数据集上IOU为0.5时mAP值可达81.43%，相较于原始CenterNet提升了3.63%.文中提出的改进方法可以在没有明显增加推理时间的前提下明显提高安全帽的检测精度.
|
| 关 键 词： | 安全帽检测 Focal-Mse-One loss Focal-Mse-Guss loss 特征融合 目标检测 |
Improvement of Helmet Detection Algorithm Aming at CenterNet Shortcomings |
| |
| WANG Hairui?,ZHAO Jianghe,WU Lei,XIE Siyuan.Improvement of Helmet Detection Algorithm Aming at CenterNet Shortcomings[J].Journal of Hunan University(Naturnal Science),2023(8):125-133. |
| |
| Authors: | WANG Hairui? ZHAO Jianghe WU Lei XIE Siyuan |
| |
| Abstract: | |
| |
| Keywords: | |
|
| 点击此处可从《湖南大学学报(自然科学版)》浏览原始摘要信息 |
| 点击此处可从《湖南大学学报(自然科学版)》下载免费的PDF全文 |
