吉林大学学报(信息科学版) ›› 2024, Vol. 42 ›› Issue (3): 559-566.
张利巍, 杨万帅
ZHANG Liwei, YANG Wanshuai
摘要: 针对加油站场景中的目标检测算法存在检测精度低的问题, 提出一种基于 Yolov3-Tiny 的加油站场景目标检测改进算法。 该算法以 Yolov3-Tiny 模型为基础网络, 引入 Yolov4 算法提出的 Mosaic 图像增强方式进行数据预处理, 采用密集连接模块重构特征提取网络, 并将 CBAM (Convolutional Block Attention Module)注意力模块与金字塔池化模块 (Pyramid Pooling Module)加入到网络中, 最终实现了加油站场景下的目标检测。 实验结果表明, 改进的算法相比于原算法的总体 mAP 提升了 8. 2% , 能更有效地应用于加油站目标检测中。
中图分类号: