基于改進(jìn)YOLOv8的電力場(chǎng)景通用缺陷檢測(cè)模型

韓睿，戴哲仁，蔣鵬，李晨，姜雄偉

（1.國網(wǎng)浙江省電力有限公司電力科學(xué)研究院，杭州 310014；2.國網(wǎng)浙江省電力有限公司，杭州 310007）

0 引言

隨著國家經(jīng)濟(jì)水平日益提高，居民及生產(chǎn)的用電需求日益加大，電網(wǎng)設(shè)備數(shù)量增多，電網(wǎng)安全穩(wěn)定運(yùn)行的重要性日漸凸顯。為保障電網(wǎng)的可靠運(yùn)行，電網(wǎng)的巡檢技術(shù)已從傳統(tǒng)人工巡檢過渡到以視頻圖像數(shù)據(jù)為媒介的自動(dòng)化巡檢，提升了電網(wǎng)巡檢的自動(dòng)化、信息化水平［1］。為提升多場(chǎng)景、多類別的電力缺陷檢測(cè)的綜合性能，基于視頻圖像分析的深度學(xué)習(xí)目標(biāo)檢測(cè)算法獲得了越來越廣泛的關(guān)注［2］。文獻(xiàn)［3-4］在雙階段目標(biāo)檢測(cè)網(wǎng)絡(luò)Faster R-CNN（改進(jìn)的快速區(qū)域卷積神經(jīng)網(wǎng)絡(luò)）的基礎(chǔ)上，實(shí)現(xiàn)了無人值守變電站視頻監(jiān)控系統(tǒng)對(duì)動(dòng)態(tài)小尺寸入侵目標(biāo)體的識(shí)別與定位。文獻(xiàn)［5-7］通過改進(jìn)單階段目標(biāo)檢測(cè)網(wǎng)絡(luò)YOLOv3，實(shí)現(xiàn)了絕緣子異物檢測(cè)。

隨著基于視頻圖像檢測(cè)的智能巡檢技術(shù)推廣，數(shù)據(jù)的采集與處理迎來了新的挑戰(zhàn)。現(xiàn)行的巡檢系統(tǒng)多采用集中式數(shù)據(jù)處理，先將各采集點(diǎn)數(shù)據(jù)回傳至數(shù)據(jù)中心，隨后再統(tǒng)一進(jìn)行數(shù)據(jù)處理與分析。隨著數(shù)據(jù)采集裝置時(shí)空分布更加分散，部分采集裝置可能存在數(shù)據(jù)傳輸信號(hào)弱，回傳速度慢甚至丟失回傳數(shù)據(jù)的情形；面對(duì)山火等突發(fā)式缺陷場(chǎng)景可能引起漏檢，錯(cuò)過最佳處理時(shí)機(jī)。此外，由于數(shù)據(jù)流量增大，數(shù)據(jù)采集點(diǎn)流量費(fèi)用增加，提高了總的運(yùn)行成本。文獻(xiàn)［8-12］指出，電力設(shè)備與控制中心、電力用戶與電力企業(yè)間的數(shù)據(jù)流大量增加，亟需在邊緣側(cè)架設(shè)邊緣終端設(shè)備對(duì)源頭數(shù)據(jù)進(jìn)行實(shí)時(shí)處理。文獻(xiàn)［13-14］指出，部分電力缺陷需要快速做出響應(yīng)，而傳統(tǒng)的檢測(cè)手段在實(shí)時(shí)性上有所不足，需要開發(fā)一種實(shí)時(shí)性、輕量化的檢測(cè)模型。

現(xiàn)有檢測(cè)模型依據(jù)物體定位與物體分類，可以分為雙階段（two-stage）檢測(cè)模型與單階段（one-stage）檢測(cè)模型。相較于雙階段模型，單階段模型具有輕量化、實(shí)時(shí)性等特點(diǎn)，但在檢測(cè)精度上不如雙階段模型，典型代表為YOLO 系列模型。本文選擇YOLOv8 作為檢測(cè)模型，以實(shí)現(xiàn)輕量化、實(shí)時(shí)性的電力缺陷檢測(cè)，并通過一系列改進(jìn)措施提升其檢測(cè)精度。此外，當(dāng)前電力領(lǐng)域缺陷檢測(cè)網(wǎng)絡(luò)［9，15-17］主要聚焦于單個(gè)缺陷，缺少一種通用的缺陷檢測(cè)模型。這種定制化開發(fā)不適用于模型的通用化部署，增加了前期投入成本與后期維護(hù)成本。為此，本文在第1章通過算法和模塊對(duì)YOLOv8 做了多種改進(jìn)，以提升其在電力場(chǎng)景的檢測(cè)精度；第2 章在14 個(gè)電力缺陷場(chǎng)景下對(duì)改進(jìn)YOLOv8 進(jìn)行測(cè)試與驗(yàn)證，取得了不錯(cuò)的實(shí)驗(yàn)效果。

1 電力場(chǎng)景缺陷檢測(cè)輕量化模型

1.1 改進(jìn)型Mosaic數(shù)據(jù)增強(qiáng)算法

目標(biāo)檢測(cè)任務(wù)的模型效果與訓(xùn)練階段的數(shù)據(jù)集規(guī)模、分布、圖像質(zhì)量有關(guān)。數(shù)據(jù)集分布涵蓋了數(shù)據(jù)集類別的分布、待檢測(cè)物體大小的分布以及數(shù)據(jù)集圖像場(chǎng)景的分布多個(gè)維度。其中，數(shù)據(jù)集圖像場(chǎng)景包括天氣環(huán)境、攝像機(jī)清晰度、背景環(huán)境多個(gè)影響因素。如果數(shù)據(jù)集圖像場(chǎng)景較為單一，則會(huì)降低模型的泛化性能，對(duì)于來自數(shù)據(jù)集外場(chǎng)景的測(cè)試數(shù)據(jù)效果較差。同時(shí)，如果豐富數(shù)據(jù)集圖像場(chǎng)景，則會(huì)對(duì)模型的識(shí)別精度有所提升。電力缺陷數(shù)據(jù)集包含多種類別的缺陷數(shù)據(jù)，不同類別缺陷所處環(huán)境具有一定差異。然而，現(xiàn)有的缺陷圖像數(shù)據(jù)集往往在同一缺陷類別上的圖像場(chǎng)景分布較為單一，降低了模型的性能，并且不利于模型的泛化，因此無法滿足實(shí)際電力安全生產(chǎn)的需求。本章提出一種適用于電力場(chǎng)景的改進(jìn)型Mosaic 算法，來豐富數(shù)據(jù)集中同一類別的場(chǎng)景，以提升模型的精度以及泛化性能。

Mosaic 算法最早在YOLOv4［18］中提出，用于通用場(chǎng)景下的目標(biāo)檢測(cè)數(shù)據(jù)集增強(qiáng)。其主要思想是先將4張來自多種類別的圖像進(jìn)行隨機(jī)裁剪、隨機(jī)翻轉(zhuǎn)、隨機(jī)縮放、隨機(jī)色調(diào)變換等前置數(shù)據(jù)預(yù)處理，再對(duì)處理后的圖像數(shù)據(jù)進(jìn)行隨機(jī)拼合，得到一張新的數(shù)據(jù)集圖像，如圖1虛線路徑所示。新的數(shù)據(jù)集圖像包含了多個(gè)類別的待檢測(cè)對(duì)象，同時(shí)對(duì)于其中一個(gè)檢測(cè)對(duì)象而言，其檢測(cè)場(chǎng)景融合了多個(gè)場(chǎng)景的特征，因而變得更加復(fù)雜多樣。Mosaic算法除了提升場(chǎng)景的多樣性外，還為數(shù)據(jù)集提供了更多的小目標(biāo)物體，解決了目標(biāo)檢測(cè)任務(wù)固有的對(duì)小目標(biāo)識(shí)別能力差的問題，在原始任務(wù)中對(duì)模型的性能有所提升。

圖1 改進(jìn)型Mosaic數(shù)據(jù)增強(qiáng)算法示意圖Fig.1 Schematic diagram of the improved Mosaic data augmentation algorithm

然而，電力場(chǎng)景的缺陷識(shí)別與通用場(chǎng)景下的物體識(shí)別存在不同，部分類別對(duì)于隨機(jī)裁剪、隨機(jī)色調(diào)變換較為敏感。如圖1右上所示，在絕緣子破損的缺陷識(shí)別中，如果破損位置被裁剪掉，那么從外觀上看是有遮擋的絕緣子，無法判別是否有破損，不應(yīng)當(dāng)被識(shí)別。相應(yīng)檢測(cè)框從正樣本變?yōu)樨?fù)樣本，引起數(shù)據(jù)標(biāo)注錯(cuò)誤。又如圖1 左下所示，呼吸器硅膠變色缺陷，往往通過硅膠筒的顏色來判斷是否出現(xiàn)異常。如果對(duì)圖像使用隨機(jī)色調(diào)變換，則顏色特征不再對(duì)模型的識(shí)別起效，從而大大降低了模型的識(shí)別能力。

為保留Mosaic 數(shù)據(jù)增強(qiáng)算法在目標(biāo)檢測(cè)任務(wù)上的各項(xiàng)優(yōu)勢(shì)，同時(shí)不破壞電力場(chǎng)景數(shù)據(jù)集的圖像特征，本文對(duì)Mosaic 算法進(jìn)行了改進(jìn)。考慮提升數(shù)據(jù)多樣性對(duì)模型訓(xùn)練的價(jià)值，本文不再限定只取4張圖像進(jìn)行合成，而是從多個(gè)類別中隨機(jī)選取2～6 張圖進(jìn)行組合。本文定義產(chǎn)生負(fù)面影響的變換與相應(yīng)類別構(gòu)成一組沖突關(guān)系，例如隨機(jī)裁剪與絕緣子破損為一組。通過分析所有缺陷類別與圖像變換的關(guān)系，整理出沖突關(guān)系表，在代碼中通過字典進(jìn)行表達(dá)，其中部分沖突關(guān)系見表1。依據(jù)沖突關(guān)系表，對(duì)各類別數(shù)據(jù)選取合適的前置數(shù)據(jù)處理，以保留圖像的關(guān)鍵信息。最后，隨機(jī)打亂圖像次序，進(jìn)行圖像拼合，如圖1 右下所示。改進(jìn)保留了Mosaic 數(shù)據(jù)增強(qiáng)算法的優(yōu)勢(shì)，同時(shí)也使其更適應(yīng)電力場(chǎng)景的缺陷檢測(cè)，從而提升輕量化檢測(cè)模型的精度。

表1 沖突關(guān)系（部分）Table 1 Conflict relationships（part）

1.2 改進(jìn)型Bottleneck模塊

Bottleneck模塊是當(dāng)前輕量化網(wǎng)絡(luò)中的常用模塊，最早在ResNet［19］中提出。Bottleneck模塊具有多個(gè)變種，YOLOv8中的Bottleneck結(jié)構(gòu)見圖2（a）。Bottleneck 模塊將輸入先經(jīng)過一個(gè)卷積核大小為3的卷積層，壓縮特征維度；再通過一個(gè)卷積核大小為3的卷積層恢復(fù)特征維度，最后使用殘差連接得到完整輸出。Bottleneck模塊主要功能是減少模型的復(fù)雜性和計(jì)算負(fù)擔(dān)，同時(shí)保持或提高模型的性能。

圖2 Bottleneck模塊與Res2Net模塊對(duì)比Fig.2 Comparison between the Bottleneck module and the Res2Net module

為進(jìn)一步提升模型對(duì)電力場(chǎng)景下多尺度特征的識(shí)別能力，本文使用Res2Net模塊［20］代 替Bottleneck模塊，在增加多尺度信息的同時(shí)保持模型的輕量化與推理速度。Res2Net的具體結(jié)構(gòu)見圖2（b）。輸入首先通過卷積核為1的卷積層，隨后將特征分成4 部分，分別為x1、x2、x3、x4。其中x1不經(jīng)過卷積直接獲得y1；x2經(jīng)過一次卷積層后獲得y2；同時(shí)將x2卷積后的特征與x3合并后再經(jīng)過卷積層獲得y3；類似的，x3 經(jīng)過卷積層的特征與x4合并后，再經(jīng)過一次卷積層得到y(tǒng)4。從總體來看，y1、y2、y3、y4 可以視作輸入的部分特征分別經(jīng)過卷積核為1、3、5、7 的卷積層，從而擁有多尺度感受野；進(jìn)一步提升了輕量化模型的檢測(cè)精度以及在多尺度目標(biāo)上的泛化能力。

1.3 電力缺陷檢測(cè)的NMS算法

目標(biāo)檢測(cè)模型會(huì)根據(jù)輸入圖像輸出一系列候選框，具體包括候選框的坐標(biāo)以及候選框的置信度大小。然而，這些候選框會(huì)出現(xiàn)交叉、重疊現(xiàn)象，即一個(gè)物體可能會(huì)被多個(gè)檢測(cè)框識(shí)別，需要算法對(duì)候選框進(jìn)行去重。NMS（非極大值抑制）算法［21］是一種常見的檢測(cè)框聚類去重算法。該算法首先按照置信度大小對(duì)候選框排序，將置信度最高的框b1作為初始選定的框；隨后計(jì)算置信度第二高的框b2與b1的IoU（重疊度）值，將兩個(gè)候選框相交面積S1除以兩個(gè)候選框并集面積S2，得：

其中S1，S2參數(shù)可視化見圖3。如果IoU值大于設(shè)定閾值（一般取0.5）則保留，否則舍棄。后續(xù)框依次與前過程中保留的候選框計(jì)算IoU 值，同樣依據(jù)是否達(dá)到閾值來決定保留與舍棄，最終實(shí)現(xiàn)去重。

圖3 IoU與CIoU參數(shù)可視化Fig.3 IoU and CIoU parameter visualization

然而，利用IoU值作為NMS算法的去重依據(jù)具有多種弊端。這是因?yàn)镮oU 只考慮了候選框的形狀重疊程度，而未考慮候選框之間的距離。意味著即使兩個(gè)候選框之間的重疊面積較小，但如果它們之間的距離很近，可能會(huì)表示為兩個(gè)不同的目標(biāo)。傳統(tǒng)基于IoU的NMS算法會(huì)將其中一個(gè)目標(biāo)框剔除，導(dǎo)致召回率下降。此外，基于IoU的NMS 算法在處理高密度目標(biāo)時(shí)容易出現(xiàn)問題。當(dāng)多個(gè)目標(biāo)框密集堆疊在一起時(shí)，它們之間的重疊面積可能非常大，導(dǎo)致基于IoU的傳統(tǒng)NMS算法無法準(zhǔn)確區(qū)分目標(biāo)框，從而選擇性地保留其中一部分目標(biāo)框，而其他目標(biāo)框則被抑制。這也會(huì)導(dǎo)致召回率下降。而對(duì)于電力場(chǎng)景的識(shí)別任務(wù)，召回率是一個(gè)相當(dāng)重要的評(píng)價(jià)指標(biāo)，關(guān)系到缺陷設(shè)備與隱患的漏檢比例，與電力的安全生產(chǎn)及穩(wěn)定運(yùn)行有重要關(guān)聯(lián)。

CIoU［22］最早用于檢測(cè)任務(wù)訓(xùn)練階段的損失函數(shù)，代替?zhèn)鹘y(tǒng)基于IoU 的損失函數(shù)，綜合考量了目標(biāo)框的尺寸、距離、重疊面積，提升目標(biāo)框的定位精度?；诖?，本節(jié)使用CIoU代替IoU作為NMS 算法的計(jì)算依據(jù)。假設(shè)待計(jì)算CIoU 值的候選框分別為b1和b2，令d為b1、b2中心點(diǎn)之間的歐式距離，c為b1、b2最小外接矩形的對(duì)角線長度，w1、w2分別為b1、b2的寬度，h1、h2分別為b1、b2的高度。相關(guān)參數(shù)可視化見圖3。則b1、b2的CIoU計(jì)算公式如下：

其中，CIoU 通過引入?yún)?shù)v，進(jìn)一步解決了IoU對(duì)候選框的長寬不敏感的問題；α為懲罰系數(shù)，決定了v在多大程度上影響CIoU的值。CIoU提升了檢測(cè)模型在聚類去重階段的召回率與精度，進(jìn)而提升了輕量化模型的綜合性能。

1.4 改進(jìn)YOLOv8整體結(jié)構(gòu)

YOLO系列模型作為單階段檢測(cè)模型的代表，具有參數(shù)規(guī)模小，檢測(cè)速度快、泛化性好等優(yōu)點(diǎn)。本文模型整體框架參考YOLOv8中的YOLOv8m，并根據(jù)1.1—1.3 節(jié)對(duì)其進(jìn)行改進(jìn)，在保留YOLOv8輕量化、實(shí)時(shí)性等優(yōu)點(diǎn)的同時(shí)，進(jìn)一步提升了模型在電力場(chǎng)景下的檢測(cè)精度與召回率，使其更符合電力生產(chǎn)的實(shí)際需求。模型在訓(xùn)練階段的結(jié)構(gòu)如圖4所示。輸入的圖像數(shù)據(jù)首先經(jīng)過數(shù)據(jù)增強(qiáng)，隨后送入特征提取網(wǎng)絡(luò)提取多尺度特征；提取的多尺度特征通過上采樣、拼接融合多尺度特征，隨后進(jìn)入C2f網(wǎng)絡(luò)，進(jìn)一步增強(qiáng)模型對(duì)不同大小物體的識(shí)別能力。隨后特征經(jīng)過幾次卷積層，獲得分類和定位的損失值，完成一次訓(xùn)練過程，經(jīng)過多輪訓(xùn)練后得到最終模型。

圖4 模型整體結(jié)構(gòu)Fig.4 The overall structure of the model

2 實(shí)驗(yàn)結(jié)果與分析

2.1 數(shù)據(jù)集和訓(xùn)練配置

本文從變電站采集了14 種類別的缺陷數(shù)據(jù)，并對(duì)數(shù)據(jù)進(jìn)行去重、篩選等工作，隨后進(jìn)行人工打標(biāo)。每類缺陷數(shù)據(jù)按照7∶3隨機(jī)劃分為訓(xùn)練集與測(cè)試集，再將各類別的訓(xùn)練集、測(cè)試集數(shù)據(jù)分別進(jìn)行合并。最終得到了圖片總量為50 177 張的實(shí)驗(yàn)數(shù)據(jù)集。實(shí)驗(yàn)數(shù)據(jù)集中各類別的名稱、訓(xùn)練集與測(cè)試集樣本數(shù)量如表2所示。部分?jǐn)?shù)據(jù)具有多類別的缺陷，例如呼吸器硅膠筒破損與呼吸器油封異?？赡艹霈F(xiàn)在同一圖像中，因此在表2中會(huì)被重復(fù)統(tǒng)計(jì)；但實(shí)際在劃分訓(xùn)練集與測(cè)試集時(shí)仍會(huì)視為同一個(gè)訓(xùn)練樣本。從數(shù)據(jù)分布角度看，各類別數(shù)據(jù)集均在2 000～5 000張內(nèi)，訓(xùn)練與測(cè)試的數(shù)據(jù)充分，且數(shù)據(jù)分布較為均勻，避免了樣本不足以及數(shù)據(jù)長尾的問題。

表2 實(shí)驗(yàn)數(shù)據(jù)集配置Table 2 Experimental dataset configuration

訓(xùn)練使用的軟硬件中：GPU 為8 張NVIDIA RTX3090，CPU 為Intel（R）Xeon（R）Gold 6242R，系統(tǒng)環(huán)境為Ubuntu16.04，深度學(xué)習(xí)框架為Pytorch。模型訓(xùn)練優(yōu)化器采用SGD（隨機(jī)梯度下降），初始學(xué)習(xí)率為0.01，動(dòng)量值為0.94，權(quán)值衰減為0.000 5，采用線性的學(xué)習(xí)率策略，batch（批）設(shè)為128，EMA（指數(shù)移動(dòng)平均）衰減設(shè)為0.999。從訓(xùn)練epoch（輪次）分別為50，100，150，200 的模型中選擇在測(cè)試集上表現(xiàn)最優(yōu)的一個(gè)作為結(jié)果。

2.2 實(shí)驗(yàn)結(jié)果

2.2.1 評(píng)價(jià)指標(biāo)

目標(biāo)檢測(cè)通常用AP（平均精度）去衡量模型在某個(gè)類別上的檢測(cè)效果，并對(duì)各類別的AP值做平均，得到模型在多類別場(chǎng)景下的性能指標(biāo)mAP。AP綜合考量了precision（準(zhǔn)確率）和recall（召回率）這兩個(gè)指標(biāo)，反映了模型在漏檢與誤檢方面的性能。AP計(jì)算時(shí)，根據(jù)模型輸出檢測(cè)框與真實(shí)框的IoU值是否達(dá)到設(shè)定閾值，將檢測(cè)框判定為預(yù)測(cè)正確或錯(cuò)誤。因此根據(jù)閾值的不同，AP也分為好幾種不同的小類，例如AP50，AP75，AP@0.5∶0.95。其中AP50 和AP75 分別代表IoU 閾值設(shè)置為50%和75%，而AP@0.5∶0.95 代表以IoU 從50%到95%的閾值范圍內(nèi)每隔5%計(jì)算一次AP，最后取平均值得到最終AP 值。因而AP 指標(biāo)本身反映了模型漏檢與誤檢方面的數(shù)值，不同的AP指標(biāo)對(duì)于檢測(cè)框的定位精度提出了不同的要求。

對(duì)于實(shí)際電力場(chǎng)景下的缺陷檢測(cè)任務(wù)，相比過高的檢測(cè)框精度，漏檢率和誤檢率更具有價(jià)值。這是因?yàn)闄z測(cè)算法主要用于缺陷的判別，后續(xù)缺陷處理主要基于人工手段，而AP50對(duì)于人工而言已經(jīng)足以定位缺陷位置。因此本文選取AP50作為評(píng)價(jià)標(biāo)準(zhǔn)，并相應(yīng)地選擇各類別的AP50 平均值mAP50作為模型在多類別時(shí)的評(píng)價(jià)指標(biāo)。

2.2.2 缺陷檢測(cè)結(jié)果

考慮電力場(chǎng)景的實(shí)際需求，本文選取平均精度AP50 作為模型的評(píng)價(jià)指標(biāo)。利用改進(jìn)的YOLOv8 模型在14 個(gè)場(chǎng)景下進(jìn)行了實(shí)驗(yàn)。為更好比較本文改進(jìn)模型的檢測(cè)效果，本文分別選取了在速度上占優(yōu)的YOLO 系列模型YOLOv8，以及在精度上占優(yōu)的Mask R-CNN［23］作為基準(zhǔn)，并用相同數(shù)據(jù)集、相同設(shè)備進(jìn)行訓(xùn)練和測(cè)試。各缺陷類別的實(shí)驗(yàn)結(jié)果對(duì)比如圖5所示?？梢钥闯霰疚母倪M(jìn)的YOLOv8模型在所有14個(gè)場(chǎng)景上均比YOLOv8模型和Mask R-CNN 有更好的檢測(cè)精度，并在9個(gè)場(chǎng)景上取得了超過90%的檢測(cè)精度。僅在掛空懸浮物、金屬銹蝕、山火3個(gè)場(chǎng)景上檢測(cè)精度低于80%，主要原因是這些場(chǎng)景數(shù)據(jù)較難標(biāo)注，訓(xùn)練效果不佳；且金屬銹蝕類本身與環(huán)境區(qū)分度小，檢測(cè)難度大。盡管如此，本文提出的模型在這3個(gè)場(chǎng)景下較原有方法均有顯著提升。

圖5 改進(jìn)YOLOv8與YOLOv8、Mask R-CNN在14個(gè)場(chǎng)景下的實(shí)驗(yàn)結(jié)果對(duì)比Fig.5 Comparison of experimental results using the improved YOLOv8，YOLOv8，and Mask R-CNN in 14 scenarios

為進(jìn)一步比較電力缺陷場(chǎng)景下本文改進(jìn)的YOLOv8 模型與現(xiàn)有模型在參數(shù)量、推理速度等指標(biāo)上的優(yōu)劣，本文在相同的硬件設(shè)施上對(duì)這些模型的指標(biāo)進(jìn)行考察，結(jié)果如表3 所示。改進(jìn)型YOLOv8 在YOLOv8 基礎(chǔ)上簡化了參數(shù)量，提升了速度，并在mAP50上仍有所提升，可見本文提出的改進(jìn)YOLOv8 模型在現(xiàn)有模型中具有速度和參數(shù)量上的優(yōu)勢(shì)。進(jìn)一步證明了本文模型在14 個(gè)電力場(chǎng)景下均提升了模型精度的同時(shí)，仍保留了單階段模型輕量化、實(shí)時(shí)性的特性。

表3 改進(jìn)YOLOv8在電力缺陷場(chǎng)景下與YOLOv8，Mask R-CNN指標(biāo)對(duì)比結(jié)果Table 3 Comparison of indices using the improved YOLOv8，YOLOv8，and Mask R-CNN in power defect scenarios

2.2.3 消融實(shí)驗(yàn)

為驗(yàn)證本文引入的改進(jìn)Mosaic算法、Res2Net模塊、CIoU-NMS 算法的有效性，進(jìn)行多組消融實(shí)驗(yàn)。首先將Mosaic 數(shù)據(jù)增強(qiáng)算法、IoU-NMS的YOLOv8 作為基準(zhǔn)；然后在基準(zhǔn)上僅替換Mosaic 數(shù)據(jù)增強(qiáng)算法為本文算法以驗(yàn)證改進(jìn)Mosaic數(shù)據(jù)增強(qiáng)對(duì)模型的影響；接著在基準(zhǔn)上僅改變NMS的計(jì)算依據(jù)為CIoU以驗(yàn)證CIoU對(duì)模型的影響。最后驗(yàn)證上述多種改進(jìn)一起加入對(duì)模型的提升作用。實(shí)驗(yàn)結(jié)果如表4所示。

表4 消融實(shí)驗(yàn)結(jié)果Table 4 Ablation Experiment Results

由表4可見，在電力場(chǎng)景下本文提出的各項(xiàng)改進(jìn)方法均對(duì)模型有一定提升。其中，改進(jìn)Mosaic算法和CIoU-NMS算法對(duì)模型的精度提升最為顯著，驗(yàn)證了本文提出的各項(xiàng)改進(jìn)方法的有效性。

3 結(jié)語

本文對(duì)適用于通用場(chǎng)景下的YOLOv8 模型進(jìn)行若干改進(jìn)，在保留其輕量化、實(shí)時(shí)性的同時(shí)，提升其在電力場(chǎng)景下的檢測(cè)精度。利用沖突關(guān)系表來改善Mosaic 數(shù)據(jù)增強(qiáng)算法在電力數(shù)據(jù)上的局限性，使用Res2Net 模塊優(yōu)化了YOLOv8 模型的輕量化與多尺度特性，結(jié)合CIoU NMS 進(jìn)一步提升了YOLOv8 模型的檢測(cè)精度。通過實(shí)驗(yàn)驗(yàn)證了改進(jìn)后的YOLOv8 模型仍具有輕量化、實(shí)時(shí)性等特點(diǎn)，同時(shí)在14 個(gè)缺陷場(chǎng)景下具有更高的檢測(cè)精度，對(duì)電力安全生產(chǎn)具有重要意義。長遠(yuǎn)來看，本文為基于邊緣計(jì)算等分布式、輕量化技術(shù)的電力智能檢測(cè)系統(tǒng)提供了理論模型，為保障電力領(lǐng)域安全生產(chǎn)提供了新的思路。