針對(duì)鋼材表面小目標(biāo)缺陷實(shí)時(shí)檢測(cè)

摘要： 鋼材表面缺陷檢測(cè)任務(wù)中，YOLO將目標(biāo)檢測(cè)轉(zhuǎn)換為對(duì)位置信息的回歸問題，實(shí)現(xiàn)高幀率實(shí)時(shí)檢測(cè)，但對(duì)小目標(biāo)缺陷定位精度有所欠缺。針對(duì)該問題，以YOLOv5s架構(gòu)為基礎(chǔ)，首先，在模型輸入端設(shè)定動(dòng)態(tài)尺度訓(xùn)練范式，提高小目標(biāo)缺陷訓(xùn)練精度；其次，設(shè)計(jì)STD-CA模塊利用圖像轉(zhuǎn)換技術(shù)，避免下采樣過程中分辨率的降低，導(dǎo)致小目標(biāo)缺陷特征信息的丟失，并引導(dǎo)特征提取能力，降低無關(guān)背景特征關(guān)注度，進(jìn)一步...