基于改進(jìn)YOLOv8n 的井下人員安全帽佩戴檢測(cè)

打開文本圖片集

摘要：針對(duì)現(xiàn)有井下人員安全帽佩戴檢測(cè)方法未考慮遮擋、目標(biāo)較小、背景干擾等因素，存在檢測(cè)精度差及模型不夠輕量化等問(wèn)題，提出一種改進(jìn)YOLOv8n 模型，并將其應(yīng)用于井下人員安全帽佩戴檢測(cè)。在頸部網(wǎng)絡(luò)中加入P2 小目標(biāo)檢測(cè)層，提高模型對(duì)小目標(biāo)的檢測(cè)能力，更好地捕捉安全帽目標(biāo)細(xì)節(jié)；在主干網(wǎng)絡(luò)中添加卷積塊注意力模塊（CBAM）提取圖像關(guān)鍵特征，減少背景信息的干擾；將CIoU 損失函數(shù)替換為WIoU 損失函數(shù)，提升模型對(duì)檢測(cè)目標(biāo)的定位能力；采用輕量化共享卷積檢測(cè)頭（LSCD），通過(guò)共享參數(shù)的方式降低模型復(fù)雜度，并將卷積中的歸一化層替換為群組歸一化（GN），在盡可能保證精度的同時(shí)實(shí)現(xiàn)模型輕量化。（剩余1728字）