计算机科学
行人检测
行人
算法
工程类
运输工程
作者
Kunyue Li,Yifan Li,Yiyi Wang,Haining Yu,Huan Zhang
出处
期刊:IEEE Access
[Institute of Electrical and Electronics Engineers]
日期:2024-01-01
卷期号:12: 184827-184840
标识
DOI:10.1109/access.2024.3512666
摘要
Accurate and efficient real-time pedestrian detection in hospitals is crucial for improving safety operations and effective management. However, this task poses significant challenges due to complex scenes, dense crowds, and pedestrian occlusions. This paper proposes a multi-branch anchor-free algorithm for hospital pedestrian detection. Firstly, a multi-branch network structure is added after the backbone network of the model to adapt to multiple key local features of pedestrian targets. Subsequently, a distance loss function between key regions is designed to guide the branch networks in learning the differential detection positions of pedestrians locally. Furthermore, using ResNet34 as the baseline feature generation network, four upsampling blocks are appended at the end to form a hourglass structure, enhancing the branch network’s understanding of spatial information in pedestrian local features. Lastly, a local feature selection network is proposed to adaptively suppress non-optimal values from the multi-branch outputs, eliminating redundant feature boxes during prediction. Experimental results demonstrate that the method achieved an AP of 89.2% on the CrowdHuman dataset, indicating high detection accuracy. Additionally, on the HospitalPerson dataset for pedestrian detection in hospitals, the F1 score, Recall, and AP reached 0.9, 94.59%, and 85.42% respectively, showcasing the superior performance of the proposed method in hospital pedestrian detection, particularly in crowded and heavily occluded scenarios.
科研通智能强力驱动
Strongly Powered by AbleSci AI