面向嵌入式的CV&深度学习-YOLO V3网络

168 阅读1分钟

感谢同济子豪兄开源贡献,可去B站搜索详细内容,本文是笔者学习总结。

1 YOLO V3论文中特征提取

  • 按照步长计算公式,得到128.5,此处是因为最后一个卷积再卷就到图片外面了,舍弃最后一个卷积,所以执行向下取整,得到128*128 image.png image.png
  • yolov3.cfg 网络结构文件 image.png
  • 总共三个yolo层,也即3种尺度,下图为最后一个尺度,用于检测小目标 image.png

2 YOLO V3目标检测迁移学习

2.1 模型文件准备

image.png image.png

2.2 迁移学习

image.png image.png image.png

  • 归一化。缩放到416*416,RGB image.png image.png
  • prediction 是包含3个元素的列表 image.png image.png image.png
  • 提取预测框信息 image.png image.png
  • (13*13+26*26+52*52)*3=10647个预测框 image.png image.png image.png image.png image.png

2.3 实时处理模板

image.png image.png image.png