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

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

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

target detectiondense connection moduleattention mechanismpyramid pool moduleimage enhancement

张利巍、杨万帅

展开 >

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

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

2024

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

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

CSTPCD
影响因子:0.607
ISSN:1671-5896
年,卷(期):2024.42(3)
  • 5