殷浤益 熊友亮 孫全 劉海濤
摘要:為提升城市排水管網(wǎng)缺陷檢測效率并降低成本,研究提出了一種基于優(yōu)化YOLOv5算法的排水管網(wǎng)缺陷自動(dòng)識(shí)別方案。在原有YOLOv5框架上,引入解耦頭方法以解決模型輸出層間的耦合性問題;針對(duì)圖像樣本模糊不清的問題,運(yùn)用簡易最優(yōu)運(yùn)輸分配(simOTA)算法優(yōu)化正負(fù)樣本匹配策略,從而增強(qiáng)檢測精確度。研究最終通過消融實(shí)驗(yàn)對(duì)改進(jìn)模型與原模型進(jìn)行了對(duì)比。結(jié)果顯示,經(jīng)過優(yōu)化的YOLOv5模型相較于基礎(chǔ)模型,在平均精度(mAP)上提高了13.2%,F(xiàn)1 得分提升了4%,并在各項(xiàng)性能評(píng)估指標(biāo)上均超越原模型,展現(xiàn)出在排水管網(wǎng)缺陷檢測領(lǐng)域的較高應(yīng)用價(jià)值。
關(guān)鍵詞:排水管網(wǎng);目標(biāo)識(shí)別;YOLOv5;解耦頭;simOTA
中圖分類號(hào):TP391 文獻(xiàn)標(biāo)識(shí)碼:A
文章編號(hào):1009-3044(2024)10-0041-05