基于改進YOLOv5的排水管網缺陷識別研究

殷浤益 熊友亮 孫全 劉海濤

摘要：為提升城市排水管網缺陷檢測效率并降低成本，研究提出了一種基于優化YOLOv5算法的排水管網缺陷自動識別方案。在原有YOLOv5框架上，引入解耦頭方法以解決模型輸出層間的耦合性問題；針對圖像樣本模糊不清的問題，運用簡易最優運輸分配（simOTA）算法優化正負樣本匹配策略，從而增強檢測精確度。研究最終通過消融實驗對改進模型與原模型進行了對比。結果顯示，經過優化的YOLOv5模型相較于基礎模型，在平均精度（mAP）上提高了13.2%，F1 得分提升了4%，并在各項性能評估指標上均超越原模型，展現出在排水管網缺陷檢測領域的較高應用價值。

關鍵詞：排水管網；目標識別；YOLOv5；解耦頭；simOTA

中圖分類號：TP391 文獻標識碼：A

文章編號：1009-3044（2024）10-0041-05