基于YOLOv7-TSCR的連鑄坯表面缺陷檢測

打開文本圖片集

摘要：為解決連鑄生產(chǎn)過程中鑄坯表面缺陷檢測準(zhǔn)確率低、檢測速度慢、模型參數(shù)量大難以部署等問題，提出一種融合重參數(shù)化和注意力機(jī)制的輕量化鑄坯表面缺陷檢測算法YOLOv7-TSCR。首先，利用Mish和SiLU激活函數(shù)、SimAM注意力機(jī)制，構(gòu)建了改進(jìn)的高效層聚合模塊ELAN-S，以有效增強(qiáng)對多尺度缺陷特征的提取；其次，設(shè)計了C2f_RG模塊改進(jìn)特征融合網(wǎng)絡(luò)，減少參數(shù)量的同時獲得更豐富的梯度流信息，增強(qiáng)特征融合能力；最后，根據(jù)采集實際生產(chǎn)中的缺陷圖像，構(gòu)建鑄坯缺陷數(shù)據(jù)集并進(jìn)行驗證。（剩余15259字）