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