电子科技大学学报2024，Vol.53Issue(3)：382-389,8.DOI:10.12178/1001-0548.2022070
联合结构重参数和YOLOv5的航拍红外目标检测
Infrared Aerial Object Detection Based on Structural Re-Parameterization and YOLOv5
摘要
Abstract
Infrared aerial object detection has been widely used in transportation,agriculture,military security,and other areas.The main challenges are small objects,mutual occlusion,little texture information,weak edge features,and large deformation of non-rigid bodies.To address these problems,based on YOLOv5 and structural Re-Parameterization(Rep),an improved object detection network Rep-YOLO is proposed for infrared aerial object detection.Firstly,the RepVGG module is introduced in the backbone network to improve the model feature extraction capability.During the model inference,the branches of the RepVGG module are structurally re-parameterized to reduce the branch and the complexity of the network structure.Secondly,the path aggregation network(PANet)in the neck of the detection network is improved by combining the priori feature,to increase the accuracy and speed balance capability.Finally,experiments are conducted on two publicly available infrared datasets,showing that the algorithm can effectively detect aerial infrared objects.Compared with the baseline(YOLOv5s),the statistical results on ComNet dataset show the mean Average Precision(mAP)is increased by 5.9%,while the parameters and model size are reduced by about 29.7%and 23.2%,respectively.In addition,the model deployment verification of our Rep-YOLO is carried out on the airborne platform Jetson Nano.It provides reliable technical support for the improvement of the detection algorithm and its practical application with UAV platforms.关键词
深度学习/红外图像/航拍目标检测/YOLOv5/结构重参数化Key words
deep learning/infrared imaging/aerial object detection/YOLOv5/structural re-parameterization分类
信息技术与安全科学引用本文复制引用
邵延华,张兴平,张晓强,楚红雨,吴亚东..联合结构重参数和YOLOv5的航拍红外目标检测[J].电子科技大学学报,2024,53(3):382-389,8.基金项目
国家自然科学基金(6160382) （6160382）
国防科工项目(20zg6108) （20zg6108）
四川省科技厅项目(2019YJ0325,2020YFG0148) （2019YJ0325,2020YFG0148）
