计算机工程与设计2024,Vol.45Issue(2) :420-427.DOI:10.16208/j.issn1000-7024.2024.02.013

改进的YoloV3轻量化多目标检测算法

Improved YoloV3 lightweight multi-target detection algorithm

苏擎凯 童莹 何睿清 曹雪虹
计算机工程与设计2024,Vol.45Issue(2) :420-427.DOI:10.16208/j.issn1000-7024.2024.02.013

改进的YoloV3轻量化多目标检测算法

Improved YoloV3 lightweight multi-target detection algorithm

苏擎凯 1童莹 2何睿清 2曹雪虹3
扫码查看

作者信息

  • 1. 南京邮电大学通信与信息工程学院,江苏南京 210003
  • 2. 南京工程学院信息与通信工程学院,江苏南京 211167
  • 3. 南京邮电大学通信与信息工程学院,江苏南京 210003;南京工程学院信息与通信工程学院,江苏南京 211167
  • 折叠

摘要

为有效解决轻量化目标检测网络降低参数量导致的检测精度损失问题,设计一种轻量化多 目标检测网络MSPF-YoloV3,可以通过低分组数、低内存占用的通道混洗结构,达到与高分组数同样的信息交流的效果;将位置信息丰富的浅层特征与深层特征进行融合,提高对不同大小目标的检测精度.经过实验验证了该方法的有效性与实时检测的能力.该网络在PASCAL VOC07+12数据集上的mAP达到86.31%;在COCO2014数据集中提取出与PASCAL VOC07+12数据集相同的20个类别,mAP达到67.71%;权重文件大小为46.8 MB,相较于YoloV3降低了 198.2 MB;检测速度达到44 FPS.

Abstract

To solve the problem of detection accuracy loss caused by reducing the number of parameters in lightweight target de-tection network,a target detection network MSPF-YOLOV3 was designed,which achieved the same information communication effect as high packet number through the channel shuffling structure with low packet number and low memory footprint.Shallow features with rich location information were fused with deep features to improve the detection accuracy of targets of different sizes.After experiments,the mAP of this network on PASCAL VOC07+12 dataset reaches 86.31%.In COCO2014 dataset,20 categories are extracted that are the same as PASCAL VOC07+12 dataset,and the mAP reaches 67.71%.The weight file size is 46.8 MB,showing a decrease of 198.2 MB compared to YoloV3.The detection speed reaches 44 FPS.The validity and the ability of real-time detection of the method are verified.

关键词

目标检测/轻量化/低分组数/低内存占用/通道混洗/特征融合/实时检测

Key words

target detection/lightweight/low number of clusters/low memory footprint/channel shuffling/feature fusion/real-time detection

引用本文复制引用

基金项目

国家自然科学基金项目(61703201)

国家自然科学基金项目(61905108)

江苏省高等学校自然科学研究面上基金项目(19KJB140010)

南京工程学院科研基金项目(ZKJ202003)

南京工程学院引进人才科研启动基金项目(YKJ201868)

出版年

2024
计算机工程与设计
中国航天科工集团二院706所

计算机工程与设计

CSTPCD北大核心
影响因子:0.617
ISSN:1000-7024
参考文献量22
段落导航相关论文