摘 要：帶鋼作為現(xiàn)代鋼鐵產(chǎn)業(yè)的核心產(chǎn)品，劃痕檢測對于確保產(chǎn)品質(zhì)量、提升生產(chǎn)效率和降低成本至關(guān)重要，廣泛應用于汽車制造、金屬加工、電子產(chǎn)品生產(chǎn)等領(lǐng)域。然而，劃痕形態(tài)各異且易受光照、背景和噪聲等因素影響，使得檢測任務極具挑戰(zhàn)性。近年來，隨著空間數(shù)據(jù)智能技術(shù)的不斷進步，基于深度學習的目標檢測算法（如Ｆａｓｔｅｒ ＲＣＮＮ、ＳＳＤ、ＹＯＬＯ 等） 在檢測任務中表現(xiàn)出色，通過自動學習、特征檢測和精準目標定位，在復雜背景下也能準確檢測?；冢伲希蹋希觯的Ｐ瓦M行了算法結(jié)構(gòu)的改進，將空間金字塔池化（Ｓｐａｔｉａｌ Ｐｙｒａｍｉｄ Ｐｏｏｌｉｎｇ，ＳＰＰ） 模塊替換為快速空間金字塔池化（ＳｐａｔｉａｌＰｙｒａｍｉｄ ＰｏｏｌｉｎｇＦａｓｔ，ＳＰＰＦ） 模塊，引入注意力機制，改進現(xiàn)有的目標檢測算法，提升劃痕檢測的準確性和魯棒性。結(jié)合卷積塊注意力機制模塊（Ｃｏｎｖｏｌｕｔｉｏｎａｌ Ｂｌｏｃｋ Ａｔｔｅｎｔｉｏｎ Ｍｏｄｕｌｅ，ＣＢＡＭ） 構(gòu)建了ＹＯＬＯｖ５ＣＢＡＭ 模型。ＣＢＡＭ 通過關(guān)注通道和空間維度上的信息，使模型更精準地聚焦于劃痕區(qū)域，提升了檢測效果。實驗結(jié)果顯示，ＹＯＬＯｖ５ＣＢＡＭ 模型在各類交并比（Ｉｎｔｅｒｓｅｃｔｉｏｎ ｏｖｅｒ Ｕｎｉｏｎ，ＩｏＵ） 閾值下相較于ＹＯＬＯｖ５，精確率、召回率和ｍＡＰ ＠ ０． ５ 有著較好的表現(xiàn)，分別提升了５． ６％ 、９． １％ 和５． ９％ 。隨著空間數(shù)據(jù)智能技術(shù)的不斷進步，未來有望為劃痕檢測提供更多創(chuàng)新思路和解決方案。

關(guān)鍵詞：劃痕檢測；ＹＯＬＯｖ５；卷積塊注意力機制模塊；模型構(gòu)建與訓練

中圖分類號：ＴＰ３１５ 文獻標志碼：Ａ

文章編號：１００３－３１０６（２０２４）１２－２７８９－１１

０ 引言

鋼鐵工業(yè)作為國家經(jīng)濟的基礎(chǔ)支撐產(chǎn)業(yè)，不僅是構(gòu)建現(xiàn)代化強國的重要基石，還扮演著推動綠色低碳發(fā)展的角色。帶鋼作為當代鋼鐵工業(yè)的核心產(chǎn)出品，伴隨著工業(yè)智能制造技術(shù)的持續(xù)進步，其市場需求正不斷攀升。然而，隨著生產(chǎn)能力的增強，市場對于帶鋼品質(zhì)的期望也越來越高。熱軋工序在帶鋼制造中扮演著核心角色，被眾多工廠采用。在熱軋帶鋼的生產(chǎn)過程中，表面劃痕的存在與否成為了衡量產(chǎn)品質(zhì)量的關(guān)鍵指標。劃痕等缺陷會顯著降低產(chǎn)品的品質(zhì)，從而影響熱軋帶鋼質(zhì)量的整體水平。

從２０ 世紀７０ 年代以來，渦流［１］、紅外線［２］等多種檢測方法開始興起。紅外檢測方法是一種以工件瞬態(tài)導熱原理為基礎(chǔ)的非破壞性測試手段［３］。從２０ 世紀９０ 年代起，人們開始對帶鋼的表面缺陷進行研究，有了一些較成熟的檢測方法。張濤等［４］提出了一種改進的層次分析法，機器視覺技術(shù)借助電荷耦合器件（Ｃｈａｒｇｅ Ｃｏｕｐｌｅｄ Ｄｅｖｉｃｅ，ＣＣＤ）相機的廣泛應用，正在逐步取代傳統(tǒng)的帶鋼表面缺陷檢測方法，發(fā)展成為識別帶鋼表面缺陷的主要技術(shù)手段［４］。在國外，許多企業(yè)較早地采用機器視覺技術(shù)來檢測帶鋼的表面瑕疵。德國的Ｐａｒｓｙｔｅｃ 公司成功研發(fā)了一種使用面陣ＣＣＤ 攝像頭的ＨＴＳ２ 帶鋼表面缺陷檢測系統(tǒng)［５］。Ｈａｎｇ 等［６］提出一種基于領(lǐng)域像素灰度閾值的缺陷檢測方法，通過缺陷及其相鄰像素的灰度情況來判斷缺陷類型，獲得了良好的效果。然而，某些缺陷可能會被誤歸為背景，進而造成分類錯誤。為了解決這一問題，提出一種基于背景差分與改進遺傳算法最大熵的軌道表面缺陷分割方法［７］。該方法首先采用改進的列灰度均值背景圖像建模技術(shù)對軌道表面圖像進行背景建模，隨后通過計算軌道表面圖像與背景圖像的差分來獲取差分圖像。其次，利用改進遺傳算法的最大熵原理來確定差分圖像的最佳分割閾值，并對其進行二值化處理。最后，對軌道表面的二值圖像進行形態(tài)學處理和濾波，從而得到軌道表面缺陷的分割圖像。這一解決方案在帶鋼表面缺陷檢測領(lǐng)域表現(xiàn)出色且相對成熟，至今仍在廣泛應用。在國內(nèi)，北京科技大學徐科等［８］采用多個面陣ＣＣＤ 攝像機同步捕獲鋼板表面圖像，并通過構(gòu)建一個由多臺客戶機與一臺服務器組成的并行計算機系統(tǒng)，成功實現(xiàn)了對鋼板表面部分缺陷的高效檢測。Ｌｉ 等［９］采用智能漩渦、漏磁記憶檢測儀等技術(shù)對曲軸外表面進行無損檢測，但是檢測成本過高且誤差過大，對小目標的檢測效果不佳。上海寶鋼集團攜手東北大學共同研發(fā)了一款熱軋鋼板表面缺陷檢測系統(tǒng)，已在寶鋼集團的生產(chǎn)車間中付諸實踐應用［１０］。

近期，計算機技術(shù)的快速進步使得利用深度學習技術(shù)識別帶鋼表面缺陷的圖像成為了研究的熱點。當前，許多關(guān)于帶鋼表面缺陷檢測的研究都集中在運用卷積神經(jīng)網(wǎng)絡（Ｃｏｎｖｏｌｕｔｉｏｎａｌ ＮｅｕｒａｌＮｅｔｗｏｒｋ，ＣＮＮ）自動進行特征提取和缺陷識別，這種方法省去了手動特征提取的環(huán)節(jié)，并且通常能夠達到比傳統(tǒng)機器學習技術(shù)更優(yōu)的檢測性能。Ｈｅ 等［１１］提出了一套創(chuàng)新的缺陷檢測框架，該框架首先對缺陷圖像進行分類，然后根據(jù)不同類別的缺陷使用特定的卷積核來提取特征，最終確定缺陷的位置。雖然這種方法在提高缺陷檢測率方面表現(xiàn)出色，但也帶來了模型復雜性增加的問題。主要分為以ＳＳＤ［１２］、ＣｅｎｔｅｒＮｅｔ［１３］、ＹＯＬＯ 系列［１４］為代表的單階段網(wǎng)絡，以及以ＲＣＮＮ［１５］、Ｆａｓｔ ＲＣＮＮ［１６］、Ｆａｓｔｅｒ ＲＣＮＮ［１７］和Ｍａｓｋ ＲＣＮＮ［１８］為代表的兩階段網(wǎng)絡。二者之間的主要區(qū)別在于是否存在生成區(qū)域候選框的階段。一階段目標檢測算法無需預先生成區(qū)域候選框。檢測結(jié)果可以通過網(wǎng)絡直接計算，速度很快，但檢測精度可能相對較低。兩階段目標檢測算法過程分為２ 個階段。首先，生成候選框，然后根據(jù)這些候選框優(yōu)化檢測點以獲得更高的準確性，但代價是檢測速度較慢。這種方法檢測精度較高，但是檢測速度較慢。其中，在２０１５ 年Ｊｏｓｅｐｈ Ｒｅｄｍｏｎ 和ＡｌｉＦａｒｈａｄｉ 首次提出ＹＯＬＯ 系列［１９］算法，其檢測速度和精度非常平衡，更適合檢測工業(yè)缺陷。東北大學的宋克臣教授團隊創(chuàng)新性地引入了生成對抗網(wǎng)絡（Ｇｅｎｅｒａｔｉｖｅ Ａｄｖｅｒｓａｒｉａｌ Ｎｅｔｗｏｒｋ，ＧＡＮ）應用于帶鋼表面缺陷檢測，采用ＧＡＮ 技術(shù)，有效緩解了帶鋼表面缺陷分類任務中樣本稀缺的問題，通過創(chuàng)建大量的未標記缺陷圖像數(shù)據(jù)來增強缺陷識別能力［２０］。這種方法顯著擴充了缺陷圖像的庫容，盡管如此，ＧＡＮ 在訓練階段相對繁瑣，且在模擬復雜場景下的缺陷時遇到了一些難題。此外，該團隊還提出了一種名為缺陷檢測網(wǎng)絡（Ｄｅｆｅｃｔ Ｄｅｔｅｃｔｉｏｎ Ｎｅｔｗｏｒｋ，ＤＤＮ）的網(wǎng)絡架構(gòu)，該架構(gòu)是對Ｆａｓｔｅｒ ＲＣＮＮ 的改進，顯著提高了模型的檢測精度，成功實現(xiàn)了端到端的帶鋼表面缺陷檢測［２１］。雖然該模型在檢測精度方面表現(xiàn)優(yōu)異，但其檢測速度較慢，尚無法滿足實時檢測的需求。季娟娟等［２２］提出了一種將注意力機制與ＹＯＬＯｖ４ 相結(jié)合的網(wǎng)絡，該網(wǎng)絡分割模型凸顯出鋼表面缺陷，識別微小的缺陷，但由于只用了空間注意力機制，特征提取能力較差。

針對現(xiàn)有的目標檢測算法仍存在一定的局限性，且對劃痕檢測任務的檢測準確性不高、對復雜背景的魯棒性不強，本文引入了基于ＹＯＬＯｖ５ 模型，將空間金字塔池化（Ｓｐａｔｉａｌ Ｐｙｒａｍｉｄ Ｐｏｏｌｉｎｇ，ＳＰＰ）模塊替換成快速空間金字塔池化（Ｓｐａｔｉａｌ ＰｙｒａｍｉｄＰｏｏｌｉｎｇＦａｓｔ，ＳＰＰＦ）模塊，并結(jié)合卷積塊注意力機制模塊（Ｃｏｎｖｏｌｕｔｉｏｎａｌ Ｂｌｏｃｋ Ａｔｔｅｎｔｉｏｎ Ｍｏｄｕｌｅ，ＣＢＡＭ）構(gòu)建了ＹＯＬＯｖ５ＣＢＡＭ 模型，通過同時關(guān)注通道和空間２ 個維度的信息，使得模型能夠更加關(guān)注劃痕區(qū)域，從而提高檢測的準確性。

１ 相關(guān)理論與技術(shù)

１． １ ＹＯＬＯｖ５

在目標檢測領(lǐng)域，基于候選區(qū)域的算法曾經(jīng)是關(guān)鍵技術(shù)。這種方法主要經(jīng)歷２ 個階段：從大量的候選位置中篩選出最合適的幾個；在這些選定的區(qū)域進行對象的識別和位置的精確調(diào)整。ＹＯＬＯ 算法則采用不同的策略，通過一個ＣＮＮ 直接在整幅圖像上識別和定位對象，將目標檢測的任務轉(zhuǎn)換為一次性的回歸問題，實現(xiàn)了目標類別和位置的同時預測。

ＹＯＬＯ 的亮點在于其快速、實時和高效的性能表現(xiàn)，能夠在速度與精確度之間找到一個優(yōu)秀的平衡點。ＹＯＬＯ 算法將圖像分割成Ｓ×Ｓ 的格子，每個格子負責預測對象的類別和位置。這種方法摒棄了傳統(tǒng)的候選區(qū)域提取步驟，通過簡化的流程加速了檢測速度。ＹＯＬＯ 檢測系統(tǒng)主要由一個基礎(chǔ)的網(wǎng)絡結(jié)構(gòu)和若干后續(xù)的卷積層組成。它常用的基礎(chǔ)網(wǎng)絡，如ＤａｒｋＮｅｔ５３，是一個事先訓練好的深卷積網(wǎng)絡。后續(xù)卷積層的任務是產(chǎn)生目標的邊框和分類標簽。在訓練階段，ＹＯＬＯ 利用交叉熵和均方誤差等多種損失函數(shù)來優(yōu)化模型，在推斷階段，則通過結(jié)合框的類別概率和置信度得分來過濾和校正最終的檢測結(jié)果。

圖像劃分網(wǎng)格如圖１ 所示［２３］。展示了眾多的三維網(wǎng)格，每個網(wǎng)格都能夠檢測和識別圖中的物體。當物體中心位于某個特定網(wǎng)格內(nèi)，該網(wǎng)格便啟動檢測程序。通過邊緣條紋技術(shù)，圖像中物體的微小細節(jié)得以更準確地捕獲。邊界框以其特有的形態(tài)和功能而顯著，其寬度（ｂｗ ）、高度（ｂｈ ）、類型（如人、車輛、紅綠燈等）以及中心點位置都能被明確地確定。邊界框如圖２ 所示，清楚展示了邊界框的示例，使用紅色輪廓來突出顯示邊界框。

ＹＯＬＯ 是一種創(chuàng)新的目標檢測方法，通過將傳統(tǒng)的目標檢測任務簡化為單一的回歸問題，并結(jié)合ＣＮＮ 來進行高效識別，顯著提升了檢測的速度和精度。ＹＯＬＯ 的架構(gòu)由多個關(guān)鍵模塊組成，這些模塊相互配合，共同提升模型性能。首先，ＹＯＬＯ 采用了殘差塊設計，不僅優(yōu)化了訓練速度，還提高了模型的準確性。其次，在目標檢測過程中，ＹＯＬＯ 通過邊框回歸為每個檢測到的目標預測獨特的邊框，提供包括目標位置和尺寸的關(guān)鍵信息。此外，ＹＯＬＯ 使用交并比（Ｉｎｔｅｒｓｅｃｔｉｏｎ ｏｖｅｒ Ｕｎｉｏｎ，ＩｏＵ）來評估預測邊框與真實邊框之間的重合程度，從而驗證預測的準確性。通過這些模塊的整合，ＹＯＬＯ 實現(xiàn)了在復雜場景下高效、準確的目標檢測能力。最終結(jié)果如圖３ 所示。

圖３ 中的圖片被分割成多個網(wǎng)格，每個網(wǎng)格能夠準確地預測出Ｂ 個邊框，并為每個邊框分配置信度得分。使用單元模型來識別每個物體的類別，特別關(guān)注汽車、狗和自行車等對象。采用多ＣＮＮ 并行處理不同的預測任務，通過綜合分析來確保預測與實際相符。此方法排除了與目標屬性不相關(guān)的邊緣條件，利用精心設計的邊框精確識別物體。圍繞汽車和自行車的是鮮明的色彩邊緣，營造出城堡般的輪廓；而狗則被深藍色的邊框明確標出，這種處理技術(shù)有效強調(diào)了目標的特征，提升了檢測的精確度和可靠性。

ＹＯＬＯｖ５ｓ 模型的整體網(wǎng)絡結(jié)構(gòu)如圖４ 所示。

ＹＯＬＯｖ５ 的架構(gòu)被劃分為Ｉｎｐｕｔ、Ｂａｃｋｂｏｎｅ、Ｎｅｃｋ 及Ｐｒｅｄｉｃｔｉｏｎ 四個關(guān)鍵部分。在Ｉｎｐｕｔ 階段，采用Ｍｏｓａｉｃ數(shù)據(jù)增強技術(shù)對輸入的圖像進行隨機裁剪和合并，旨在提升模型對新場景的適應性。Ｂａｃｋｂｏｎｅ 階段為特征提取的核心，決定了模型的整體性能。ＹＯＬＯｖ５ 運用跨階段局部網(wǎng)絡（Ｃｒｏｓｓ Ｓｔａｇｅ Ｐａｒｔｉａｌ Ｎｅｔｗｏｒｋ，ＣＳＰＮｅｔ）架構(gòu)對特征圖進行切割和處理，一部分經(jīng)過卷積層，另一部分進行下采樣，之后合并這兩部分，增加了對非線性特征的識別能力，改善了模型對復雜環(huán)境和多樣目標的檢測效率。在Ｎｅｃｋ 階段，采用Ｃ３ 卷積塊進一步融合特征圖。在Ｐｒｅｄｉｃｔｉｏｎ 階段，處理后的特征圖被用來精準預測目標的位置和尺寸。

１． ２ ＣＢＡＭ

注意力機制（Ａｔｔｅｎｔｉｏｎ Ｍｅｃｈａｎｉｓｍ）是機器學習領(lǐng)域內(nèi)一種廣泛應用的數(shù)據(jù)處理技術(shù)，它覆蓋了自然語言處理、圖像識別、語音識別等多種機器學習任務。這一機制通過賦予不同信息以不同的權(quán)重（代表其重要程度），實現(xiàn)對信息的差異化關(guān)注。具體而言，注意力機制可被視為由查詢矩陣（Ｑｕｅｒｙ）、鍵（Ｋｅｙ）以及通過計算這些元素得到的加權(quán)平均值所構(gòu)成的多層感知器（Ｍｕｌｔｉｌａｙｅｒ Ｐｅｒｃｅｐｔｒｏｎ，ＭＬＰ ）結(jié)構(gòu)。

ＣＢＡＭ 是一種旨在提升ＣＮＮ 性能的注意力機制模塊［２４］，通過引入通道注意力和空間注意力來提高模型的感知能力，從而在不增加網(wǎng)絡復雜性的情況下改善性能。ＣＢＡＭ 的主要目標是克服傳統(tǒng)ＣＮＮ 在處理不同尺度、形狀和方向信息時的局限性［２５］。為此，ＣＢＡＭ 引入了２ 種注意力機制：通道注意力和空間注意力機制。

ＣＢＡＭ 是一種簡單有效的前饋ＣＮＮ 的注意力模塊，給出一個中間的特征圖，然后ＣＢＡＭ 會按照２ 個獨立的維度（通道和空間）依次推導出注意圖示，然后用自適應特征優(yōu)化乘以輸入特征圖推導出注意圖示，將其與輸入特征圖相乘來推導出注意圖示。經(jīng)過這２ 個注意力模塊的共同作用后，得到最終的注意力增強特征圖，傳遞給網(wǎng)絡的下一層進行進一步處理。這種注意力機制有助于提高網(wǎng)絡在視覺任務上的表現(xiàn)。

ＣＢＡＭ 的應用范圍廣泛，已被應用于各種領(lǐng)域，如目標檢測、圖像分割、圖像分類等。通過引入ＣＢＡＭ，很多模型在各種任務中都取得了顯著的性能提升。