SAC-RSM: A High-Performance UAV-Side Road Surveillance Model Based on Super-Resolution Assisted Learning
Efficiently and precisely identifying small items on traffic highways using unmanned aerial vehicle (UAV) platforms with limited resources is a crucial yet challenging job. This research suggests a speedy, accurate, and component-optimized road surveillance model (SAC-RSM) for UAVs. This model addresses slow detection speed, limited detection of small objects, and deployment difficulties. First, we designed a super-resolution-assisted learning branch in the network to balance the model’s detection speed and accuracy. This branch learns the feature representation from low to high resolution. This branch uses multiscale feature fusion in the encoding stage to enhance the feature representation of small objects, thereby enhancing their detection accuracy. Second, to avoid the problem of cross-layer convolution, which results in the loss of fine-grained information and low-learning efficiency, we propose using the convolution-to-space-convolution (CSPC) module in the backbone network to improve model detection’s robustness. Third, to achieve real-time detection, we realized the model using the Huawei Ascend compute architecture for neural networks (CANNs) framework to enable automatic quantization and parallel inference acceleration. Finally, we deployed the accelerated model to the embedded platform Atlas 200I developer kit (DK) A2. Compared to the baseline model, the proposed method shows significant increases in mean average precision (mAP) values for the VisDrone and DroneVehicle data sets, with increases of 17.4% and 9.4%, respectively. The proposed method achieves frames/s (FPS) of 38.3, which is 2.1 times faster than the baseline model, meeting the requirement for high-performance real-time detection in a UAV environment.
Authors
- He Xu (ORCID: https://orcid.org/0000-0003-2809-2237)
- Peng Li (ORCID: https://orcid.org/0000-0001-5026-5347)
- Wenlong Zheng (ORCID: https://orcid.org/0009-0003-5976-7388)
- Xing Shao (ORCID: https://orcid.org/0000-0001-8148-9984)
- Ruchuan Wang
Institutions
- Nanjing University of Posts and Telecommunications (CN)
- Yancheng Institute of Technology (CN)
Publication Details
- Journal
- IEEE Internet of Things Journal
- Published
- 2024-08-05
- DOI
- https://doi.org/10.1109/jiot.2024.3438459
- Citations
- 9
- Primary Topic
- Advanced Measurement and Detection Methods
- Type
- article
- Field-Weighted Citation Impact
- 1.88
Funders
- National Natural Science Foundation of China
- Nanjing University of Posts and Telecommunications
- Six Talent Peaks Project in Jiangsu Province
- National Key Research and Development Program of China