重庆理工大学学报(自然科学) ›› 2021, Vol. 35 ›› Issue (10): 136-143.

• 信息·计算机 • 上一篇    下一篇

融合跨阶段局部网络和空间金字塔池化的YOLOV3目标检测算法

梁礼明,钱艳群,吴媛媛   

  1. 江西理工大学 电气工程与自动化学院
  • 出版日期:2021-11-09 发布日期:2021-11-09

  • Online:2021-11-09 Published:2021-11-09

摘要: 实际的检测任务中往往存在复杂的环境与背景,容易造成边界框的定位不够准确、 对各尺度的目标检测精度不高等情况。为了进一步提高 Yolov3的检测精度,本文在原算法的 基础上提出一种融合跨阶段局部网络和空间金字塔池化的 Yolov3目标检测算法。首先,将主 干网络融合跨阶段局部网络 CSPNet来提高网络的学习能力;其次,引入一种改进的空间金字塔 池化结构增强网络的局部区域特征;最后,在损失函数中融入 focalloss解决正负样本不均衡带 来的问题。实验表明:改进的 Yolov3算法在检测性能上优于原算法。

关键词: Yolov3, 跨阶段局部网络, 空间金字塔池化, focalloss, 损失函数

中图分类号: 

  • tp391