基于YOLOv5s的多場景安全帽檢測算法
摘要: 針對目前無法實(shí)時(shí)準(zhǔn)確地監(jiān)測工人是否佩戴安全帽的問題,提出一種基于YOLOv5s的多場景安全帽佩戴檢測算法(YOLOv5s-SDSNR)。首先,在YOLOv5s基礎(chǔ)上采用最優(yōu)傳輸理論將局部匹配策略改進(jìn)為全局匹配策略,增加正樣本的數(shù)量,使模型更有針對性訓(xùn)練;然后,利用解耦檢測頭對分類和定位進(jìn)行解耦,分別提升分類和定位的準(zhǔn)確性;最后,使用結(jié)構(gòu)重參數(shù)化將主干網(wǎng)絡(luò)的訓(xùn)練和推理等效轉(zhuǎn)換,以... (共8頁)
開通會員,享受整站包年服務(wù)