+高级检索
针对CenterNet缺点的安全帽检测算法改进
DOI:
作者:
作者单位:

作者简介:

通讯作者:

基金项目:


Improvement of Helmet Detection Algorithm Aming at CenterNet Shortcomings
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
    摘要:

    为解决安全帽数据集上识别率较低的问题,本文提出了一种基于改进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%.文中提出的改进方法可以在没有明显增加推理时间的前提下明显提高安全帽的检测精度.

    Abstract:

    To solve the problem of low recognition rates on helmet dataset, this paper proposes a detection method based on an improved CenterNet network structure. To tackle the problem of poor prediction results in the multi-class classification of CenterNet, this paper attempts to improve the loss function. Therefore, Focal-Mse-One loss and Focal-Mse-Guss loss are proposed and compared with the original Focal loss; Aiming at the problem of low reusability of feature map in the reasoning process of CenterNet, ASFF and DASFF structures are proposed and compared. The experimental results show that the reasoning speed can reach 20.78 frames on GeForce GTX 1050 graphics card. When the IOU is 0.5, the mAP can reach 81.43% on the helmet dataset, which is 3.63% higher than the original CenterNet’s mAP. The improved method proposed in this paper can significantly improve the detec tion accuracy of safety helmet without a significant increase in reasoning time.

    参考文献
    相似文献
    引证文献
文章指标
  • PDF下载次数:
  • HTML阅读次数:
  • 摘要点击次数:
  • 引用次数:
引用本文

王海瑞 ?,赵江河 ,吴蕾 ,谢思远.针对CenterNet缺点的安全帽检测算法改进[J].湖南大学学报:自然科学版,2023,(8):125~133

复制
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2023-08-29
  • 出版日期:
作者稿件一经被我刊录用,如无特别声明,即视作同意授予我刊论文整体的全部复制传播的权利,包括但不限于复制权、发行权、信息网络传播权、广播权、表演权、翻译权、汇编权、改编权等著作使用权转让给我刊,我刊有权根据工作需要,允许合作的数据库、新媒体平台及其他数字平台进行数字传播和国际传播等。特此声明。
关闭