基于改進(jìn)YOLOv7的火災(zāi)目標(biāo)檢測(cè)算法
燃燒科學(xué)與技術(shù)
頁(yè)數(shù): 9 2024-08-15
摘要: 提出基于改進(jìn)YOLOv7的火災(zāi)目標(biāo)檢測(cè)算法,以目標(biāo)檢測(cè)網(wǎng)絡(luò)YOLOv7為基本網(wǎng)絡(luò),采用改進(jìn)后的輕量化網(wǎng)絡(luò)EfficientNetV2-S替換其主干網(wǎng)絡(luò),從而降低模型的參數(shù)量.引入坐標(biāo)注意力機(jī)制,對(duì)EfficientNetV2-S的卷積模塊進(jìn)行改進(jìn).對(duì)損失函數(shù)進(jìn)行改進(jìn),提升火災(zāi)識(shí)別的精確率.標(biāo)注火災(zāi)目標(biāo)檢測(cè)數(shù)據(jù)集共6 150張對(duì)模型進(jìn)行訓(xùn)練,與常見火災(zāi)目標(biāo)檢測(cè)算法對(duì)比驗(yàn)證,并進(jìn)行...