An Enhanced Feature-Fusion Network for Small-Scale Pedestrian Detection on Edge Devices
Small-scale pedestrian detection is one of the challenges in general object detection. Factors such as complex backgrounds, long distances, and low-light conditions make the image features of small-scale pedestrians less distinct, further increasing the difficulty of detection. To address these challenges, an Enhanced Feature-Fusion YOLO network (EFF-YOLO) for small-scale pedestrian detection is proposed. Specifically, this method employs a backbone based on the FasterNet block within YOLOv8n, which is designed to enhance the extraction of spatial features while reducing redundant operation. Furthermore, the gather-and-distribute (GD) mechanism is integrated into the neck of the network to realize the aggregation and distribution of global information and multi-level features. This not only strengthens the faint features of small-scale pedestrians but also effectively suppresses complex background information, thereby improving the accuracy of small-scale pedestrians. Experimental results indicate that EFF-YOLO achieves detection accuracies of 72.5%, 72.3%, and 91% on the three public datasets COCO-person, CityPersons, and LLVIP, respectively. Moreover, the proposed method reaches a detection speed of 50.7 fps for 1920 × 1080-pixel video streams on the edge device Jetson Orin NX, marking a 15.2% improvement over the baseline network. Thus, the proposed EFF-YOLO method not only boasts high detection accuracy but also demonstrates excellent real-time performance on edge devices.
Authors
- Yaorong Zhang
- Xuefeng Wu (ORCID: https://orcid.org/0000-0003-2133-220X)
- Min Hu (ORCID: https://orcid.org/0000-0001-7332-5808)
- Huijun Xue
- Hao Lv (ORCID: https://orcid.org/0000-0002-8116-6592)
- Shipeng Han (ORCID: https://orcid.org/0000-0003-1168-8683)
- Teng Jiao
- Jianguo Luo
Institutions
- Xi'an University of Architecture and Technology (CN)
- Air Force Medical University (CN)
Publication Details
- Journal
- Sensors
- Published
- 2024-11-15
- DOI
- https://doi.org/10.3390/s24227308
- Citations
- 5
- Primary Topic
- Advanced Neural Network Applications
- Type
- article
- Field-Weighted Citation Impact
- 1.25