基于改進(jìn)YOLOv5算法和DeepSort算法的多目標(biāo)檢測(cè)和跟蹤
濟(jì)南大學(xué)學(xué)報(bào)(自然科學(xué)版)
頁數(shù): 8 2024-03-08
摘要: 針對(duì)因受水面波紋、反光及目標(biāo)外觀特征相似而導(dǎo)致的游泳池中目標(biāo)檢測(cè)跟蹤困難的問題,提出一種基于改進(jìn)YOLOv5算法和DeepSort算法的多目標(biāo)檢測(cè)和跟蹤方法;通過引入注意力機(jī)制改進(jìn)YOLOv5算法,增強(qiáng)算法對(duì)目標(biāo)特征的提取能力;將檢測(cè)結(jié)果輸入到DeepSort算法中,在級(jí)聯(lián)匹配中引入K鄰域限制篩選目標(biāo)檢測(cè)框,減少因目標(biāo)外觀特征不明顯引起的身份切換問題;利用匈牙利算法對(duì)檢測(cè)框和預(yù)...