基于YOLOv4-tiny算法的電網(wǎng)缺陷檢測方法、裝置、存儲介質(zhì)及電子設(shè)備

基本信息

申請?zhí)?/td> CN202110478465.8 申請日 -
公開(公告)號 CN113239994A 公開(公告)日 2021-08-10
申請公布號 CN113239994A 申請公布日 2021-08-10
分類號 G06K9/62(2006.01)I;G06N3/04(2006.01)I;G06N3/08(2006.01)I 分類 計算;推算;計數(shù);
發(fā)明人 熊勇良;郭寶明;羅志勇;秦一休;陳岱庚 申請(專利權(quán))人 廣州優(yōu)飛信息科技有限公司
代理機(jī)構(gòu) - 代理人 -
地址 510000廣東省廣州市天河區(qū)黃埔大道中311號自編2-15、16棟(僅限辦公用途)(不作廠房使用)
法律狀態(tài) -

摘要

摘要 本申請實(shí)施例公開了一種基于YOLOv4?tiny算法的電網(wǎng)缺陷檢測方法、裝置、存儲介質(zhì)及電子設(shè)備,屬于目標(biāo)檢測領(lǐng)域,該方法包括:基于熱重啟隨機(jī)梯度下降SGDR算法,使用樣本集對改進(jìn)型YOLOv4?tiny網(wǎng)絡(luò)結(jié)構(gòu)進(jìn)行訓(xùn)練得到目標(biāo)檢測網(wǎng)絡(luò);其中,在訓(xùn)練過程中采用mixup算法對所述樣本集進(jìn)行數(shù)據(jù)增強(qiáng);所述改進(jìn)型YOLOv4?tiny網(wǎng)絡(luò)結(jié)構(gòu)包括空間金字塔池化SPP層,所述SPP層用于進(jìn)行特征融合;基于目標(biāo)檢測網(wǎng)絡(luò)對待驗(yàn)證電網(wǎng)圖像進(jìn)行檢測得到電網(wǎng)缺陷檢測結(jié)果,可以實(shí)現(xiàn)電網(wǎng)缺陷的快速準(zhǔn)確的檢測。