基于改進YOLOX的輕量級鋼軌表面缺陷檢測算法

摘要： 針對現(xiàn)有基于深度學(xué)習(xí)的鋼軌表面缺陷檢測方法在嵌入式檢測系統(tǒng)上兼容性較差、計算資源占用高以及檢測速度慢的問題，提出了一種基于改進YOLOX的輕量級鋼軌表面缺陷檢測算法。模型中主干特征層以MobileNetv3單元為基礎(chǔ)，在保留其網(wǎng)絡(luò)輕量化的同時進行局部優(yōu)化，改進了淺層網(wǎng)絡(luò)的激活函數(shù)，嵌入了SE(Squeeze and Excitation)注意力機制；在加強特征層優(yōu)化了尾部的冗余...