首页|改进卷积注意力机制的轻量级检测无人机目标模型

改进卷积注意力机制的轻量级检测无人机目标模型

An improved lightweight drone detection model with a convolutional attention mechanism

扫码查看
利用深度学习中特征提取的优势,提出一种改进算法,结合改进的卷积注意力模块,将YO-LOv5模型骨干网络替换为改进的MobileNetv2轻量化网络,形成I-YOLOv5算法,旨在提高检测精确度和小目标、多目标的检测能力,同时保持实时性.为构建数据集,通过网络搜索和自主录制无人机视频的方式,用Label Img工具完成标注.结果表明,I-YOLOv5算法在检测精度上有显著提升,对小目标和多目标的检测效果更优秀,在视频检测方面表现出色,具有较好的实时性能.通过模型结构优化,使检测模型的大小减少为原来的18.6%,检测速度提升120%.I-YOLOv5算法的平均精度均值达到97.8%.
An improved algorithm was proposed.By means of the advantages of feature extraction in deep learning,combined with an improved convolutional block attention module,and with the YOLOv5 model backbone network replaced with an improved MobileNetv2 lightweight network,the I-YOLOv5 algo-rithm was formed to improve the accuracy,small target and multi-target detection capabilities while real-time performance was maintained.In order to build the data set,the Label Img tool was used to complete the annotation work by means of network search and independent recording of unmanned aerial vehicle video.Experiments showed that the improved I-YOLOv5 algorithm had a significantly better ability in detection accuracy compared with the original YOLOv5,and the effect of small target and multi-target detection was also better.The optimized algorithm performed well in video detection and had a better re-al-time performance.Through the optimization of the model structure,the size of the detection model was less than 18.6%of the original,and the detection speed increased by 120%.The average accura-cy mean of the improved I-YOLOv5 algorithm reached 97.8%.

unmanned aerial vehicleobject detectionYOLOv5 modelconvolutional block attention modulelightweight

彭艺、李睿、杨青青、凃馨月

展开 >

昆明理工大学信息工程与自动化学院,昆明 650031

昆明理工大学云南省计算机技术应用重点实验室,昆明 650500

无人机 目标检测 YOLOv5模型 卷积注意力机制 轻量化

2024

兰州大学学报(自然科学版)
兰州大学

兰州大学学报(自然科学版)

CSTPCD北大核心
影响因子:0.855
ISSN:0455-2059
年,卷(期):2024.60(4)