基于YOLOv3-tiny的二輪車頭盔檢測
計(jì)算機(jī)應(yīng)用與軟件
頁數(shù): 6 2024-05-12
摘要: 針對二輪車駕乘人員頭盔佩戴問題,提出一種基于YOLOv3-tiny的輕量化頭盔檢測模型。將原始模型主干網(wǎng)絡(luò)進(jìn)行輕量化處理,減少檢測模型的參數(shù)量,在網(wǎng)絡(luò)中添加U型特征二次融合模塊,引入關(guān)于邊框距離的DIoU損失函數(shù),用于提高檢測模型的特征提取能力和識別精度。在測試集上的實(shí)驗(yàn)表明,改進(jìn)后的模型相比原YOLOv3-tiny模型表現(xiàn)出更高的查全率和mAP及F1指標(biāo),且在保持較小參數(shù)量的... (共6頁)