FPN-CenterNet安全帽佩戴檢測算法
【摘要】:安全帽作為施工場所工人的安全保障,佩戴與否影響著工人的生命安全。在佩戴檢測方面引入深度學習可以高效地提醒工人佩戴安全帽。但由于施工圖像中安全帽的圖像過小,CenterNet表現的并不好。因此針對這個情況,提出了FPN-CenterNet框架;然后使用ACNet非對稱卷積核來對主干網絡的特征提取進行增強;最后使用DIoU損失函數來優化邊框預測的準確度。最終修改的算法相較于原始的CenterNet算法mAp提升了4.99%,在GTX GeForce 1050的GPU上的FPS達到25.81。實驗表明修改之后的算法在安全帽佩戴檢測上有良好的準確性和效率。
【相似文獻】 | ||
|
|||||||||||||||||||||||||||||||||
|
|
|||||||||
|
|
|||||||||||||||||||||||||||||||||||||||||
|
|
|||||||||||||||||||||||||||||||||||||||||
|