QQ交談
專利類型:
發(fā)明公開
申請(專利)號:
CN202411184170.X
申請日:
2024-08-27
申請公布號:
CN119180785A
申請公布日:
2024-12-24
申請人:
淮陰工學(xué)院
地址:
223005 江蘇省淮安市經(jīng)濟(jì)技術(shù)開發(fā)區(qū)枚乘東路1號
發(fā)明人:
王媛媛; 尹彤彤; 陳秀川; 劉乙麟; 張澳; 丁政; 朱葉萌; 張海燕; 楊丹
專輯:
信息科技
專題:
計(jì)算機(jī)軟件及計(jì)算機(jī)應(yīng)用
主分類號:
G06T7/00
分類號:
G06T7/00;G06V10/44;G06V10/52;G06V10/774;G06V10/80;G06V10/82;G06N3/045
國省代碼:
32
頁數(shù):
18
代理機(jī)構(gòu):
淮安市科文知識產(chǎn)權(quán)事務(wù)所
代理人:
李鋒
1.一種基于SBA跨尺度特征融合的鋼鐵缺陷檢測方法,其特征在于,所述方法包括如下步驟:S1:構(gòu)建鋼鐵缺陷數(shù)據(jù)集并對其進(jìn)行預(yù)處理:S2:以原始模型RT-DETR-R34為基準(zhǔn),改進(jìn)并獲取新鋼鐵缺陷檢測模型;S2.1:以一層Transformer的Encoder構(gòu)建RT-DETR-R34,替換backbone為starnet,利用星運(yùn)算實(shí)現(xiàn)高維和非線性特征空間的映射;S2.2:構(gòu)建新頸部結(jié)構(gòu)SCCFM,將原始模型CCFM頸部結(jié)構(gòu)Fusion模塊替換為SBA模塊,增加一條融合分支,3個SBA模塊共進(jìn)行四次特征融合,每個SBA模塊后面額外加入一個RepC3模塊,學(xué)習(xí)更豐富圖像特征;S2.3:選擇inner loss替換原有邊界框回歸損失函數(shù);S3:設(shè)置設(shè)置訓(xùn)練超參數(shù),使用S2中的訓(xùn)練集對改進(jìn)后的新鋼鐵缺陷檢測模型進(jìn)行訓(xùn)練并設(shè)置驗(yàn)證參數(shù)進(jìn)行驗(yàn)證;S4:將驗(yàn)證后的模型部署到可視化界面實(shí)現(xiàn)鋼鐵缺陷的實(shí)時檢測。
本發(fā)明公開了一種基于SBA跨尺度特征融合的鋼鐵缺陷檢測方法,首先構(gòu)建RT-DERT-R34網(wǎng)絡(luò),該網(wǎng)絡(luò)采用了一層Transformer的Encoder,來處理主干網(wǎng)絡(luò)輸出的特征。其次,本發(fā)明在提取特征階段,將骨干網(wǎng)絡(luò)替換為StarNet,通過星運(yùn)算在保持計(jì)算復(fù)雜度較低的同時,實(shí)現(xiàn)了高維特征空間的映射,無需復(fù)雜的特征融合和多分支設(shè)計(jì)就能實(shí)現(xiàn)更豐富的特征表示;然后,為了更好的捕捉多尺度特征,提升高維和低維特征的融合效果,我們設(shè)計(jì)了一種結(jié)合SBA(Selective Boundary Aggregation)的CCFM頸部結(jié)構(gòu)并命名為SCFFM(SBA-based Cross-scale Feature Fusion Module);最后,選擇inner loss作為邊界框回歸損失函數(shù)。與現(xiàn)有技術(shù)相比,本發(fā)明進(jìn)一步提高了對鋼鐵缺陷檢測精度,同時端到端檢測方式易于部署,避免了NMS帶來的延遲。