結(jié)合Transformer和動態(tài)特征融合的低照度目標(biāo)檢測
計算機(jī)工程與應(yīng)用
頁數(shù): 7 2024-01-05
摘要: 針對現(xiàn)有低照度目標(biāo)檢測算法模型參數(shù)量與計算量大、檢測實(shí)時性差、難以應(yīng)用于移動設(shè)備等問題,提出一種基于YOLOv8的改進(jìn)輕量模型DarkYOLOv8的低照度目標(biāo)檢測方法。使用MobileNet v2替換YOLOv8的主干網(wǎng)絡(luò),增加模型的特征提取能力;使用Transformer注意力機(jī)制,獲取圖像的全局信息,并且基于目標(biāo)標(biāo)記信息作為標(biāo)簽訓(xùn)練Transformer模塊參數(shù),增強(qiáng)目標(biāo)區(qū)... (共7頁)