吉林大学学报(信息科学版)2024,Vol.42Issue(3) :559-566.

基于改进Yolov3-Tiny的加油站目标检测算法研究

Research on Gas Station Target Detection Algorithm Based on Improved Yolov3-Tiny

张利巍 杨万帅
吉林大学学报(信息科学版)2024,Vol.42Issue(3) :559-566.

基于改进Yolov3-Tiny的加油站目标检测算法研究

Research on Gas Station Target Detection Algorithm Based on Improved Yolov3-Tiny

张利巍 1杨万帅1
扫码查看

作者信息

  • 1. 东北石油大学物理与电子工程学院,黑龙江大庆 163318
  • 折叠

摘要

针对加油站场景中的目标检测算法存在检测精度低的问题,提出一种基于Yolov3-Tiny的加油站场景目标检测改进算法.该算法以Yolov3-Tiny模型为基础网络,引入Yolov4算法提出的Mosaic图像增强方式进行数据预处理,采用密集连接模块重构特征提取网络,并将CBAM(Convolutional Block Attention Module)注意力模块与金字塔池化模块(Pyramid Pooling Module)加入到网络中,最终实现了加油站场景下的目标检测.实验结果表明,改进的算法相比于原算法的总体mAP提升了 8.2%,能更有效地应用于加油站目标检测中.

Abstract

We present an improved target detection algorithm based on Yolov3-Tiny for gas station scene because of the low accuracy of target detection algorithm in gas station scenes.This algorithm takes Yolov3-Tiny model as the basic network,innovates Mosaic image enhancement method proposed in Yolov4 algorithm for data preprocessing,uses dense connection modules to reconstruct the feature extraction network,and adds CBAM(Convolutional Block Attention Module)attention mechanism and Pyramid Pooling Module into the network,finally target detection in the gas station scene is realized.The experimental results show that the improved algorithm improves the overall mAP by 8.2%compared with the original algorithm,and can be more effectively applied to gas station target detection.

关键词

目标检测/密集连接模块/注意力机制/金字塔池化模块/图像增强

Key words

target detection/dense connection module/attention mechanism/pyramid pool module/image enhancement

引用本文复制引用

出版年

2024
吉林大学学报(信息科学版)
吉林大学

吉林大学学报(信息科学版)

CSTPCD
影响因子:0.607
ISSN:1671-5896
参考文献量5
段落导航相关论文