计算机工程2024,Vol.50Issue(10):51-60,10.DOI:10.19678/j.issn.1000-3428.0068711
基于热力图预测的免"锚框"人物目标检测算法
Anchor-Free Person Target Detection Algorithm Based on Heat Map Prediction
摘要
Abstract
Traditional target detection algorithms usually preset a number of candidate target bounding boxes(referred to as anchor boxes)to exhaustively cover potential target locations.Afterward,they filter out redundant bounding boxes to determine the target's location by calculating the confidence scores of the remaining boxes.This detection method requires complex post-processing and has low detection efficiency.To address these issues,this study proposes an anchor-free target detection algorithm based on heat map prediction.The algorithm transforms the detection of the character target into the detection of the maximum value on a character heat map,identifying the target's center point.By regressing the target size from the center point,the final target position can be determined,eliminating the need for anchor boxes and their associated calculations.This approach effectively reduces computational costs and significantly improves detection speed.Experimental results show that,compared to traditional anchor-based detection algorithms like Faster R-CNN and SSD,the proposed algorithm increases target detection speed to 45 frame/s.Additionally,detection accuracy on various datasets is significantly improved over the previous methods.In realistic scenes tests,the algorithm can track and accurately detect character positions in real-time,even in scenarios with character occlusions,achieving the goal of real-time detection.关键词
目标检测/免"锚框"/热力图/中心点检测/hourglass网络Key words
target detection/anchor-free/heat map/center point detection/hourglass network分类
信息技术与安全科学引用本文复制引用
王子豪,方成,李丽萍,鹿存跃..基于热力图预测的免"锚框"人物目标检测算法[J].计算机工程,2024,50(10):51-60,10.基金项目
国家自然科学基金面上项目(11174206) (11174206)
上海交通大学"深蓝计划"重点项目(SL2020ZD103). (SL2020ZD103)