摘 要:針對(duì)目前絕緣子缺陷檢測(cè)方法速度較低、網(wǎng)絡(luò)復(fù)雜度高且小目標(biāo)缺陷難以準(zhǔn)確檢測(cè)的問(wèn)題,提出一種輕量化絕緣子缺陷檢測(cè)模型———P-YOLOv7-tiny。對(duì)主干網(wǎng)絡(luò)的高效層聚合網(wǎng)絡(luò)(Efficient Layer Aggregation Network,ELAN) 模塊進(jìn)行輕量化處理,設(shè)計(jì)出PELAN 模塊以減少模型參數(shù)量,提高模型檢測(cè)速度;將坐標(biāo)注意力(Coordinate Attention,CA)機(jī)制與原模型結(jié)構(gòu)CSPSPP 進(jìn)行融合,設(shè)計(jì)出CACSPSPPS 模塊使模型更加關(guān)注絕緣子缺陷特征,提高對(duì)缺陷的檢測(cè)準(zhǔn)確率;采用定位損失函數(shù)(WIoUv3 Loss) 計(jì)算損失,將較小的梯度增益分配給低質(zhì)量的錨框以減少有害梯度,提高模型的定位性能。實(shí)驗(yàn)結(jié)果表明,P-YOLOv7-tiny 可以快速準(zhǔn)確地檢測(cè)缺陷,其中mAP@ 0. 5 達(dá)到了98. 3% ,召回率達(dá)到了95. 3% ,模型參數(shù)量為3. 1 M,計(jì)算量為7. 0 GFLOPs。相較原模型YOLOv7-tiny,所提模型適合部署到邊緣設(shè)備對(duì)絕緣子缺陷進(jìn)行實(shí)時(shí)檢測(cè)。
關(guān)鍵詞:絕緣子;缺陷檢測(cè);輕量化;YOLOv7
中圖分類(lèi)號(hào):TM769 文獻(xiàn)標(biāo)志碼:A
文章編號(hào):1003-3106(2024)10-2305-10