摘" 要:針對(duì)鋼材焊縫表面缺陷檢測(cè)中因缺陷尺度小、形態(tài)多變及缺陷與背景對(duì)比度低等因素導(dǎo)致的誤檢率與漏檢率偏高問(wèn)題,提出一種基于改進(jìn)YOLOv8模型的輕量化焊縫表面缺陷檢測(cè)方法。首先,在模型主干網(wǎng)絡(luò)引入空間金字塔分解(SPD)模塊,以增強(qiáng)模型對(duì)小尺度缺陷的檢測(cè)能力;其次,在特征融合網(wǎng)絡(luò)嵌入SimAM注意力機(jī)制,強(qiáng)化模型對(duì)低對(duì)比度缺陷的特征表征能力;再次,采用Wise-IoU替代傳統(tǒng)邊界框回歸損失函數(shù),優(yōu)化模型定位精度;最后,通過(guò)ADown模塊改進(jìn)下采樣方法,有效保留焊縫缺陷的細(xì)節(jié)特征。實(shí)驗(yàn)結(jié)果表明:改進(jìn)模型的檢測(cè)精度、召回率與平均精度均值(mAP)分別提升了3.7%、1.6%和3.6%,其綜合性能優(yōu)于原始模型及其他主流目標(biāo)檢測(cè)模型,為工業(yè)場(chǎng)景下的焊縫缺陷檢測(cè)系統(tǒng)部署提供了有效解決方案。
關(guān)鍵詞:焊縫缺陷檢測(cè);YOLOv8;SPD模塊;SimAM注意力
中圖分類號(hào):TP183;TP391.4" 文獻(xiàn)標(biāo)識(shí)碼:A" 文章編號(hào):2096-4706(2025)03-0044-06
Lightweight Weld Surface Defect Detection Method Based on Improved YOLOv8
CHEN Xin, XU Hui, ZHANG Zimian, XIONG Tiejun, CHEN Zengxiong
(Hunan Branch of China UNICOM, Changsha" 410014, China)
Abstract: Aiming at the problems of high 1 detection rate and missed detection rate in steel weld surface defect detection due to factors such as small defect scale, variable morphology and low contrast between defects and background, a lightweight weld surface defect detection method based on improved YOLOv8 model is proposed. Firstly, the Spatial Pyramid Decomposition (SPD) module is introduced into the model backbone network to enhance the model's ability to detect small-scale defects. Secondly, the SimAM is embedded in the feature fusion network to enhance the feature representation ability of the model for low contrast defects. Thirdly, Wise-IoU is used to replace the traditional bounding box regression loss function to optimize the localization accuracy of the model. Finally, the down-sampling method is improved by the ADown module to effectively retain the detailed features of the weld defects. The experimental results show that the detection accuracy, recall rate and mean Average Precision (mAP) of the improved model are increased by 3.7%, 1.6% and 3.6%, respectively. Its comprehensive performance is better than the original model and other mainstream object detection models, which provides an effective solution for the deployment of weld defect detection systems in industrial scenarios.
Keywords: weld defect detection; YOLOv8; SPD module; SimAM
0" 引" 言
鋼材作為鐵、碳等元素的合金金屬材料,在現(xiàn)代社會(huì)的各行各業(yè)中占據(jù)著極其重要的地位,應(yīng)用廣泛。鋼材表面焊縫缺陷檢測(cè)是對(duì)焊接質(zhì)量的檢驗(yàn),目的是保證產(chǎn)品結(jié)構(gòu)的完整性、可靠性和安全性[1]。同時(shí),缺陷檢測(cè)數(shù)據(jù)的積累也為焊接工藝的優(yōu)化提供了寶貴反饋,有助于持續(xù)改進(jìn)焊接質(zhì)量[2]。綜上所述,鋼材焊縫缺陷檢測(cè)對(duì)于提高鋼材產(chǎn)品質(zhì)量、減少安全風(fēng)險(xiǎn)、降低人工和材料成本、優(yōu)化焊接工藝等方面均具有重要意義。
近年來(lái),機(jī)器視覺(jué)技術(shù)和目標(biāo)檢測(cè)技術(shù)日益成熟,為鋼材焊縫缺陷檢測(cè)注入了新質(zhì)生產(chǎn)力?;谑謩?dòng)特征提取的焊縫缺陷檢測(cè)存在需要手動(dòng)設(shè)計(jì)特征提取器且適應(yīng)性差的問(wèn)題[3]。以RCNN[4]、FastRCNN[5]和Faster-RCNN[6]為代表的兩階段檢測(cè)模型,先在圖像中提取出一系列感興趣區(qū)域,然后對(duì)這些區(qū)域進(jìn)行分類和邊界框回歸。兩階段模型檢測(cè)精度良好,但速度相對(duì)較慢。為滿足實(shí)時(shí)檢測(cè)場(chǎng)景的需求,以SSD[7]和YOLO[8]系列為代表的一階段檢測(cè)算法應(yīng)運(yùn)而生。它們直接在模型中同時(shí)預(yù)測(cè)目標(biāo)物體的類別和位置,無(wú)須生成感興趣區(qū)域,因此檢測(cè)速度更快。
在焊縫缺陷檢測(cè)場(chǎng)景中,Chen等人[9]提出了一種基于YOLOv3的軋鋼表面缺陷檢測(cè)方法,采用輕量級(jí)的MobileNetV2作為特征提取器以提升檢測(cè)速度。Han等人[10]提出了一種新穎的兩階段邊緣復(fù)用網(wǎng)絡(luò)(TSERNet),該網(wǎng)絡(luò)由預(yù)測(cè)和細(xì)化兩個(gè)結(jié)構(gòu)組成,并結(jié)合缺陷的邊緣特征,有效增強(qiáng)了帶鋼表面缺陷的表征能力。Xu等人[11]提提出了改進(jìn)YOLOv7的管道焊縫表面缺陷檢測(cè)模型,設(shè)計(jì)了一個(gè)Le-HorBlock模塊,并引入了坐標(biāo)注意力機(jī)制,以減少焊縫缺陷的漏檢。李闖等人[12]提出了PCYOLOv7-tiny方法,該方法在YOLOv7-tiny模型的基礎(chǔ)上融合了可變形卷積和輕量化卷積,有效提升了檢測(cè)速度。楊杉等人[13]使用形變卷積和空洞卷積改進(jìn)MaskRCNN模型,使局部圖像能夠獲取全局圖像特征信息,能更好地提取不規(guī)則缺陷的特征。唐茂俊等人[14]采用FPN多尺度結(jié)構(gòu)和DCR解耦分類細(xì)化結(jié)構(gòu)改進(jìn)Faster-RCNN模型,以增強(qiáng)對(duì)小目標(biāo)缺陷的檢測(cè)能力。吳忍等人[15]將基于IoU值的三層級(jí)聯(lián)結(jié)構(gòu)與FasterR-CNN結(jié)合進(jìn)行焊縫缺陷檢測(cè)。
目前,基于深度學(xué)習(xí)和計(jì)算機(jī)視覺(jué)的鋼材焊縫缺陷檢測(cè)方法取得了一定成果,但在模型的精度和速度上仍有待優(yōu)化。本文以鋼材焊縫表面缺陷為研究對(duì)象,基于目前性能優(yōu)異的YOLOv8模型對(duì)缺陷檢測(cè)方法進(jìn)行改進(jìn),研究出一種輕量化和精度平衡的YOLO-WD模型,以期滿足生產(chǎn)環(huán)境中實(shí)時(shí)檢測(cè)的要求。
1" 改進(jìn)的YOLOv8方法
在YOLOv8模型[16-17]的基礎(chǔ)上,對(duì)模型結(jié)構(gòu)進(jìn)行優(yōu)化,使其更適用于復(fù)雜焊縫缺陷的快速準(zhǔn)確檢測(cè)。具體改進(jìn)方法如下:在YOLOv8模型的骨干網(wǎng)絡(luò)中引入SPD模塊[18],在頸部網(wǎng)絡(luò)中,引入SimAM注意力機(jī)制[19],使用ADown[20]替換原有的下采樣方法;替換邊界框回歸損失函數(shù)為Wise-IoU[21];這些改進(jìn)有效增強(qiáng)了模型對(duì)小目標(biāo)缺陷、形狀多變?nèi)毕莸奶卣魈崛∧芰?,在保持模型輕量化的情況下,提升了模型的檢測(cè)精度。改進(jìn)的YOLOv8模型結(jié)構(gòu)如圖1所示。
1.1" SPD模塊
在本文所采集的焊縫表面缺陷數(shù)據(jù)集中,飛濺類別的缺陷目標(biāo)尺寸較小,使用YOLOv8基礎(chǔ)模型進(jìn)行預(yù)測(cè)時(shí),發(fā)現(xiàn)存在漏檢問(wèn)題。因此,在對(duì)YOLOv8模型進(jìn)行改進(jìn)時(shí),將SPD模塊添加到其骨干網(wǎng)絡(luò)中,以彌補(bǔ)圖像中小尺寸缺陷特征信息提取不足的缺陷。
SPD模塊是SPD-Conv的重要組成部分,其結(jié)構(gòu)如圖2所示。具體來(lái)說(shuō),SPD-Conv由一個(gè)空間到深度層(Space-to-depth, SPD)和一個(gè)無(wú)步長(zhǎng)卷積層(Non-strided Conv)組成。它通過(guò)完全摒棄步長(zhǎng)卷積和池化操作,來(lái)提升對(duì)小目標(biāo)的檢測(cè)效果。
在傳統(tǒng)的CNN結(jié)構(gòu)中,由于卷積層和池化層的使用,存在目標(biāo)細(xì)粒度信息丟失的問(wèn)題。在解決鋼材焊縫表面缺陷檢測(cè)問(wèn)題時(shí),由于部分焊縫缺陷尺寸較小,CNN中的池化層和大步長(zhǎng)卷積可能導(dǎo)致目標(biāo)細(xì)節(jié)信息丟失,致使小尺寸缺陷被漏檢或誤檢。而SPD模塊將空間信息轉(zhuǎn)換為通道信息,保留了輸入特征中的所有信息。這意味著即使在網(wǎng)絡(luò)深層,也能夠保持高分辨率的特征圖,有利于精確定位和識(shí)別小尺寸的焊縫缺陷。
1.2" 嵌入SimAM注意力模塊
在鋼材焊縫表面缺陷檢測(cè)任務(wù)過(guò)程中,存在缺陷與背景物體對(duì)比度較低的情況。為了使模型能夠自適應(yīng)地著重關(guān)注特征不明顯的缺陷目標(biāo),本文采用自注意力機(jī)制模塊(SimAM),以此增強(qiáng)模型的特征提取能力,降低背景物體的干擾。在不增加模型參數(shù)量的基礎(chǔ)上,提高焊縫缺陷檢測(cè)的精度,并通過(guò)消融實(shí)驗(yàn)來(lái)驗(yàn)證 SimAM 模塊的有效性。
SimAM通過(guò)對(duì)特征圖進(jìn)行自適應(yīng)加權(quán),從而提高網(wǎng)絡(luò)對(duì)關(guān)鍵特征的關(guān)注程度。如式(1)~(3)所示,它通過(guò)最小化能量函數(shù)來(lái)計(jì)算鋼材焊縫特征圖中每個(gè)神經(jīng)元的權(quán)值。第i個(gè)神經(jīng)元的最小能量函數(shù)為。其中t表示目標(biāo)神經(jīng)元,x表示相鄰神經(jīng)元,λ表示超參數(shù),表示所有神經(jīng)元在單通道上的均值,表示所有神經(jīng)元在單通道上的方差。計(jì)算式如下:
(1)
(2)
(3)
(4)
式中,的值越小表示焊縫特征圖目標(biāo)神經(jīng)元與其他神經(jīng)元的區(qū)分度越高,即表明該神經(jīng)元越重要,特征圖上每個(gè)神經(jīng)元的權(quán)重等于1/。輸出特征圖X '如式(4)所示,E為鋼材焊縫特征圖所有神經(jīng)元 值集合,而Sigmoid激活函數(shù)的作用是限制E值過(guò)大。SimAM的整體結(jié)構(gòu)如圖3所示。首先,將原始特征圖輸入SimAM注意力機(jī)制進(jìn)行自適應(yīng)加權(quán)。然后,使用Sigmoid做權(quán)重歸一化,并將計(jì)算出的神經(jīng)元權(quán)重與原始特征圖相乘。最終,得到加權(quán)的輸出特征圖。
1.3" 下采樣模塊Adown
ADown模塊是YOLOv9模型中的重要組成部分,其主要用于特征提取和下采樣操作。如圖4所示,ADown模塊將輸入特征圖切分為兩部分,使用兩個(gè)卷積層并行處理,從而提高計(jì)算效率。接著,該模塊通過(guò)平均池化和最大池化對(duì)特征圖進(jìn)行下采樣,這不僅可以減少模型的參數(shù)量和計(jì)算量,還能有效保留重要的特征信息。通過(guò)在YOLOv8的頸部網(wǎng)絡(luò)中引入ADown模塊,模型的輕量化得以實(shí)現(xiàn),同時(shí)焊縫缺陷檢測(cè)的精度也得到了提升。
1.4" 損失函數(shù)改進(jìn)
在焊縫缺陷檢測(cè)任務(wù)中,存在許多形狀多變的缺陷,并且不同類別樣本存在不平衡的問(wèn)題。因此,優(yōu)化對(duì)形狀多變?nèi)毕莸臋z測(cè)也是本項(xiàng)研究的核心工作之一。YOLOv8模型的邊界框回歸損失函數(shù)采用的是CIoU Loss[22],計(jì)算式為:
(5)
式中,b和bgt分別表示預(yù)測(cè)框和真實(shí)框的中心點(diǎn),ρ表示兩框中心點(diǎn)的歐式距離,α表示權(quán)重平衡函數(shù),v表示框的長(zhǎng)寬比度量函數(shù),c表示預(yù)測(cè)框和真實(shí)框的最小外接矩形的對(duì)角線長(zhǎng)度。
CIoU損失函數(shù)在判斷長(zhǎng)寬比大、形狀不規(guī)則的物體時(shí)存在一定局限性,并且未考慮不同質(zhì)量樣本之間的平衡問(wèn)題。在焊縫缺陷檢測(cè)中,由于焊縫缺陷與背景的對(duì)比度較弱,且目標(biāo)尺寸差異較大,因此,本文采用WIoU損失函數(shù)替換CIoU函數(shù),以處理目標(biāo)邊界模糊以及不規(guī)則目標(biāo)的檢測(cè)難題。WIoU引入了動(dòng)態(tài)非單調(diào)聚焦機(jī)制,能夠較好地解決質(zhì)量較好與質(zhì)量較差樣本間的懲罰平衡問(wèn)題。這使得WIoU能夠聚焦于正常質(zhì)量的邊界框,更精準(zhǔn)地定位和識(shí)別圖像中的多尺度目標(biāo),進(jìn)而提升模型的檢測(cè)性能。其計(jì)算式為:
(6)
(7)
(8)
式中,(x,y)和(xgt,ygt)分別表示預(yù)測(cè)框和真實(shí)框的中心點(diǎn)坐標(biāo);Wg和Hg表示最小外接矩形框的寬和高,上標(biāo)*表示將Wg和Hg從計(jì)算圖中分離出來(lái),以防止RWIoU產(chǎn)生阻礙收斂的梯度。LIoU∈[0,1]表示IoU損失函數(shù);RWIoU∈[1,exp]表示W(wǎng)ise-IoU的懲罰項(xiàng),用于加大普通質(zhì)量錨框的LIoU;表示一個(gè)批次內(nèi)LIoU的平均值;r表示梯度增益系數(shù),用于聚焦普通質(zhì)量的錨框;α、δ表示超參數(shù),控制β和r的映射,針對(duì)不同數(shù)據(jù)集,可調(diào)整α和δ;β表示離群度,用于描述錨框的質(zhì)量,離群度小則錨框質(zhì)量高。
2" 實(shí)驗(yàn)與分析
2.1" 數(shù)據(jù)集處理
目前,在鋼材焊縫表面缺陷檢測(cè)領(lǐng)域,公開(kāi)的圖像數(shù)據(jù)較為稀缺,尚未有相對(duì)全面的公開(kāi)數(shù)據(jù)集。本文的實(shí)驗(yàn)數(shù)據(jù)集主要來(lái)源于GitHub以及百度、谷歌等平臺(tái),共收集了1 152張焊縫表面缺陷圖像。圖像中的焊縫缺陷包含飛濺、凹坑、接口不良、缺損和未融合這5個(gè)類別,如圖5所示。使用LabelImg工具進(jìn)行缺陷標(biāo)注,根據(jù)以往的模型訓(xùn)練經(jīng)驗(yàn),將數(shù)據(jù)集按照8∶2的比例劃分為訓(xùn)練集和測(cè)試集,其中訓(xùn)練集包含921張圖像,測(cè)試集包含231張圖像。
2.2" 實(shí)驗(yàn)環(huán)境和參數(shù)設(shè)置
本文中所有實(shí)驗(yàn)均在64位CentOS 7.4操作系統(tǒng)下開(kāi)展,所使用的GPU為NVIDIATeslaT4,顯存為16 GB。編程語(yǔ)言采用Python,基于PyTorch 2.0.0框架進(jìn)行模型訓(xùn)練,借助Cuda 11.7進(jìn)行加速計(jì)算,Torchvision版本為0.15.0。模型訓(xùn)練時(shí)的參數(shù)設(shè)置如表1所示。
2.3" 消融實(shí)驗(yàn)
為驗(yàn)證本文改進(jìn)模型的有效性,對(duì)改進(jìn)模型開(kāi)展了消融實(shí)驗(yàn)。以YOLOv8n模型作為基準(zhǔn)模型,將精確率(P)、召回率(R)、平均精確率均值(mAP)、模型權(quán)重(Weights)和推理時(shí)間(T)作為性能評(píng)估指標(biāo),結(jié)果如表2所示。
由表2可知,試驗(yàn)2在原始YOLOv8n模型的骨干網(wǎng)絡(luò)中加入SPD模塊后,模型的精確率、召回率和mAP分別提升了0.4%、1.8%和1.8%,而模型權(quán)重僅增加了0.3 MB,推理時(shí)間僅增加了0.6 ms。
試驗(yàn)3在YOLOv8n的頸部網(wǎng)絡(luò)中引入SimAM模塊,相較于試驗(yàn)1,雖然召回率有所降低,但精確率和mAP分別提高了5.5%和0.2%,模型大小和推理時(shí)間基本保持不變。
試驗(yàn)4將SPDConv模塊和SimAM模塊進(jìn)行組合,其中模型的召回率和mAP相較于試驗(yàn)3分別提升了3.5%和2.1%,而模型權(quán)重僅增加0.3 MB。此外,單張圖像的推理時(shí)間為10.4 ms,達(dá)到了實(shí)時(shí)檢測(cè)速度。
試驗(yàn)5在試驗(yàn)4的基礎(chǔ)上,將原始YOLOv8n模型的回歸損失函數(shù)修改為WIoU,相比試驗(yàn)4,模型的mAP和權(quán)重大小保持不變,召回率增加了1.1%,雖然精確率有所下降,但提升了模型的邊界框回歸能力。
試驗(yàn)6在試驗(yàn)5的基礎(chǔ)上,引入ADown模塊替換原有的下采樣方法。相較于試驗(yàn)5,模型的精確率和mAP分別提升了1.7%和1.3%。此外,模型權(quán)重減少了0.2 MB,推理時(shí)間減少了0.1 ms。最終,將試驗(yàn)6的模型簡(jiǎn)稱為YOLO-WD。
綜上所述,本文改進(jìn)的YOLO-WD模型在保持模型輕量化的前提下,減少了焊縫缺陷檢測(cè)的漏檢和誤檢情況,并且實(shí)現(xiàn)了較高的推理速度。
2.4" 模型對(duì)比實(shí)驗(yàn)
將本文改進(jìn)的YOLO-WD模型與YOLOv5n、YOLOv7-Tiny、YOLOv9-T和YOLOv10等主流檢測(cè)模型進(jìn)行定量對(duì)比,實(shí)驗(yàn)結(jié)果如表3所示。
根據(jù)表3的試驗(yàn)結(jié)果可知,YOLOv8n在參數(shù)量及模型權(quán)重大小均小于YOLOv7-Tiny和YOLOv9-T這兩個(gè)單階段模型的情況下,其精確率、召回率和mAP均優(yōu)于YOLOv7-Tiny和YOLOv9-T;YOLOv5n的召回率和mAP略高于YOLOv8n,然而YOLOv5n的推理時(shí)間更長(zhǎng);YOLOv10n模型的參數(shù)量、計(jì)算量和推理時(shí)間優(yōu)于YOLOv8n,但在精確率、召回率和mAP方面均不及YOLOv8n;YOLOv10s模型的精確率和mAP略高于YOLOv8n,但召回率稍顯不足。此外,該模型的參數(shù)量、計(jì)算量較多,輕量化程度欠佳。本文改進(jìn)模型YOLO-WD的精確率、召回率和mAP均優(yōu)于YOLOv8n以及其他主流目標(biāo)檢測(cè)模型。相比于YOLOv8n,雖然模型的參數(shù)量和推理時(shí)間稍有增加,但依然實(shí)現(xiàn)了模型的輕量化以及良好的實(shí)時(shí)推理速度。通過(guò)以上分析可以看出,本研究提出的YOLO-WD模型在多個(gè)指標(biāo)方面均展現(xiàn)出優(yōu)越性。
2.5" 可視化分析
為驗(yàn)證本文改進(jìn)的YOLO-WD模型在實(shí)際場(chǎng)景中的性能,選取了包含中小目標(biāo)、前景與背景對(duì)比度弱等不同場(chǎng)景的缺陷圖像進(jìn)行測(cè)試,檢測(cè)結(jié)果如圖6所示。通過(guò)直觀對(duì)比發(fā)現(xiàn),由于YOLO-WD引入了能夠保留圖像通道維度中所有信息的SPD結(jié)構(gòu),應(yīng)用自注意力機(jī)制模塊(SimAM),并且采用輕量化的下采樣模塊ADown,將邊界框回歸損失函數(shù)替換為WIoU,相比于YOLOv8n模型,YOLO-WD捕捉到了更多的焊縫缺陷特征信息。在面對(duì)小尺寸缺陷、對(duì)比度弱缺陷的情況時(shí),減少了對(duì)缺陷的漏檢和誤檢。綜合來(lái)看,對(duì)于不同尺寸的鋼材焊縫缺陷檢測(cè),改進(jìn)的YOLO-WD模型呈現(xiàn)出更優(yōu)的檢測(cè)水平。
3" 結(jié)" 論
本研究以鋼材焊縫表面存在的多類缺陷為研究對(duì)象,提出一種輕量化的焊縫表面缺陷檢測(cè)算法,實(shí)現(xiàn)了焊縫缺陷的快速檢測(cè)。在YOLOv8n的基礎(chǔ)上,將SPD模塊引入骨干網(wǎng)絡(luò),以提升小目標(biāo)的檢測(cè)精度;在YOLOv8n的頸部網(wǎng)絡(luò)中引入SimAM注意力機(jī)制,在未增加模型參數(shù)量的情況下,提升了模型的精確度;使用Wise-IoU作為損失函數(shù),提升了模型的邊界框回歸能力;采用ADown輕量化下采樣模塊,既可以減少模型參數(shù)量,又可以進(jìn)一步細(xì)化焊縫缺陷特征。最終,各類缺陷的精確率、召回率和平均精確率均值分別提高了3.7%、1.6%和3.6%。
本研究建立的模型在收集的5類焊縫缺陷檢測(cè)任務(wù)中取得了良好效果,但由于焊縫表面缺陷種類繁多、同類缺陷差異大,且實(shí)驗(yàn)樣本不夠豐富。因此,下一步的研究工作將豐富焊縫表面缺陷的種類和樣本數(shù)量,提升模型的泛化性。同時(shí),將模型進(jìn)行量化加速后,部署在邊緣設(shè)備上,以增加模型的應(yīng)用價(jià)值。
參考文獻(xiàn):
[1] 李亞森,李曄,李趙輝.基于深度學(xué)習(xí)的焊縫缺陷檢測(cè)方法綜述 [J].焊接技術(shù),2024,53(4):6-13.
[2] 劉政達(dá).復(fù)雜焊縫外觀質(zhì)量自動(dòng)檢測(cè)及評(píng)估方法研究 [D].長(zhǎng)春:吉林大學(xué),2024.
[3] 李蕾.基于機(jī)器視覺(jué)的管道內(nèi)焊縫表面缺陷識(shí)別技術(shù)研究 [D].西安:西安石油大學(xué),2023.
[4] DONG C,ZHANG K,XIE Z Y,et al. An Improved Cascade RCNN Detection Method for Key Components and Defects of Transmissi Onlines [J].IETGeneration, Transmission amp; Distribution,2023,17(19):4277-4292.
[5] WANG X L,SHRIVASTAVA A,GUPTA A. A-Fast-RCNN: Hard Positive Generation via Adversary for Object Detection [C]//2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).Honolulu:IEEE,2017:3039-3048.
[6] LIU R,YU Z H,MO D L,et al. An Improved Faster-RCNN Algorithm for Object Detection in Remote Sensing Images [C]//2020 39th Chinese Control Conference(CCC).Shenyang:IEEE,2020:7188-7192.
[7] ZHAI S P,SHANG D R,WANG S H,et al. DF-SSD: An Improved SSD Object Detection Algorithm Based on DenseNet and Feature Fusion [J]. IEEE Access,2020,8:24344-24357.
[8] JIANG P Y,ERGU D,LIU F Y,et al. A Review of YOLO Algorithm Developments [J].Procedia Computer Science,2022,199:1066-1073.
[9] CHEN X C,LYU J,F(xiàn)ANG Y L,et al. Online Detection of Surface Defects Based on Improved YOLOV3 [J].Sensors,2022,22(3):817.
[10] HAN C J,LI G Y,LIU Z. Two-Stage Edge Reuse Network for Salient Object Detection of Strip Steel Surface Defects [J].IEEE Transactions on Instrumentation and Measurement,2022,71:1-12.
[11] XU X Q,LI X. Research on Surface Defect Detection Algorithm of Pipeline Weld Based on YOLOv7 [J/OL].Scientific Reports,2024,14(1):1881[2024-07-16].https://link.springer.com/article/10.1038/s41598-024-52451-3.
[12] 李闖,馬行,穆春陽(yáng),等.改進(jìn)YOLOv3的輕量級(jí)鑄件焊縫表面缺陷檢測(cè) [J].組合機(jī)床與自動(dòng)化加工技術(shù),2024(1):156-159+163.
[13] 楊彬,亞森江·木沙,安波.改進(jìn)Mask RCNN的焊縫缺陷檢測(cè) [J].機(jī)械設(shè)計(jì)與制造,2023(6):157-161.
[14] 唐茂俊,黃海松,張松松,等.改進(jìn)的Faster-RCNN在焊縫缺陷檢測(cè)中的應(yīng)用 [J].組合機(jī)床與自動(dòng)化加工技術(shù),2021(12):83-86.
[15] 吳忍,孫淵.級(jí)聯(lián)結(jié)構(gòu)與Faster R-CNN相結(jié)合的焊縫缺陷檢測(cè) [J].組合機(jī)床與自動(dòng)化加工技術(shù),2022(2):59-62+67.
[16] 李茂,肖洋軼,宗望遠(yuǎn),等.基于改進(jìn)YOLOv8模型的輕量化板栗果實(shí)識(shí)別方法 [J].農(nóng)業(yè)工程學(xué)報(bào),2024,40(1):201-209.
[17] 李先旺,賀歲球,賀德強(qiáng),等.基于改進(jìn)YOLOv8的地鐵列車焊縫缺陷輕量化檢測(cè)方法 [J].廣西大學(xué)學(xué)報(bào):自然科學(xué)版,2024,49(3):540-552.
[18] SUNKARA R,LUO T. No More Strided Convolutions or Pooling: A New CNN Building Block for Low-Resolution Images and Small Objects [C]//Machine Learning and Knowledge Discovery in Databases.Grenoble:Springer,2022:443-459.
[19] YANG L X,ZHANG R Y,LI L D,et al. SimAM: A Simple, Parameter-Free Attention Module for Convolutional Neural Networks [C]//Proceedings of the International Conference on Machine Learning.Jeju Island:PMLR,2021:11863-11874.
[20] WANG C-Y,YEH I-H,LIAO H-Y Mark. YOLOv9: Learning What You Want to Learn Using Programmable Gradient Information [J/OL].arXiv:2402.13616 [cs.CV].[2024-07-12].https://doi.org/10.48550/arXiv.2402.13616.
[21] TONG Z J,CHEN Y H,XU Z W,et al. Wise-IoU: Bounding Box Regression Loss With Dynamic Focusing Mechanism [J/OL].arXiv:2301.10051 [cs.CV].[2024-05-26].https://doi.org/10.48550/arXiv.2301.10051.
[22] ZHENG Z H,WANG P,LIU W,et al. Distance- IoU Loss: Faster and Better Learning for Bounding Box Regression [J/OL].arXiv:1911.08287 [cs.CV].[2024-05-13].https://doi.org/10.48550/arXiv.1911.08287.
作者簡(jiǎn)介:陳新(1974—),男,漢族,湖南新化人,首席技術(shù)總監(jiān),工程師,碩士,研究方向:5G網(wǎng)絡(luò)創(chuàng)新應(yīng)用、工業(yè)互聯(lián)網(wǎng);通信作者:張孜勉(1990—),男,漢族,湖南常德人,高級(jí)技術(shù)總監(jiān),博士,研究方向:計(jì)算機(jī)視覺(jué)、圖像處理與識(shí)別。