改進(jìn)YOLOv8的輕量化軸承缺陷檢測(cè)算法
計(jì)算機(jī)工程與應(yīng)用
頁(yè)數(shù)： 10 2024-08-16
摘要： 針對(duì)軸承表面缺陷檢測(cè)存在小目標(biāo)檢測(cè)精度低、模型復(fù)雜的問(wèn)題，提出一種改進(jìn)YOLOv8的輕量級(jí)軸承缺陷檢測(cè)算法YOLO-SSW。在主干網(wǎng)絡(luò)中添加3-D注意力機(jī)制SimAM，在不增加參數(shù)的前提下，使模型更加關(guān)注軸承表面缺陷特征的提取和表達(dá)；在頸部網(wǎng)絡(luò)中嵌入C2f＿SCConv模塊，以減少空間維度和通道維度上的特征冗余，降低模型的計(jì)算負(fù)載；添加新的小目標(biāo)檢測(cè)層用于檢測(cè)軸承表面的小尺寸缺... （共10頁(yè)）
APP 