YOLOv5结构讲解

YOLOv5关闭wandb

首先打开utils/loggers/wandb/wandb_utils.py

在32行新添加wandb = None

YOLOv5结构讲解_第1张图片

然后打开utils/loggers/__init__.py

在25行新添加wandb = None

YOLOv5结构讲解_第2张图片

YOLOv5-6.0中各模型区别

yolov5n, yolov5s, yolov5m, yolov5l, yolov5x的模型只在模型深度(C3模块数)和宽度(网络的通道数)上不同,其他都一样。

以yolov5l为基准,其他模型以此放缩。

进入目录model/ ,命令行输入:grep "multiple" *.yaml

yolov5l.yaml:depth_multiple: 1.0  # model depth multiple
yolov5l.yaml:width_multiple: 1.0  # layer channel multiple
yolov5m.yaml:depth_multiple: 0.67  # model depth multiple
yolov5m.yaml:width_multiple: 0.75  # layer channel multiple
yolov5n.yaml:depth_multiple: 0.33  # model depth multiple
yolov5n.yaml:width_multiple: 0.25  # layer channel multiple
yolov5s.yaml:depth_multiple: 0.33  # model depth multiple
yolov5s.yaml:width_multiple: 0.50  # layer channel multiple
yolov5x.yaml:depth_multiple: 1.33  # model depth multiple
yolov5x.yaml:width_multiple: 1.25  # layer channel multiple
depth width
yolov5n 0.33 0.25
yolov5s 0.33 0.5
yolov5m 0.67 0.75
yolov5l 1 1
yolov5x 1.33 1.25

YOLOv5-6.0对比YOLOv5-5.0

https://github.com/ultralytics/yolov5/releases/tag/v6.0https://github.com/ultralytics/yolov5/releases/tag/v6.0

Model Architecture: Updated backbones are slightly smaller, faster and more accurate.

  • Replacement of Focus() with an equivalent Conv(k=6, s=2, p=2) layer (#4825 by @thomasbi1) for improved exportability
  • New SPPF() replacement for SPP() layer for reduced ops (#4420 by @glenn-jocher)
  • Reduction in P3 backbone layer C3() repeats from 9 to 6 for improved speeds
  • Reorder places SPPF() at end of backbone
  • Reintroduction of shortcut in the last C3() backbone layer
  • Updated hyperparameters with increased mixup and copy-paste augmentation

 在backbone有更改。YOLOv5结构讲解_第3张图片

SPPF对比SPP

位于文件:models/common.py

YOLOv5结构讲解_第4张图片

 两者结果是一样,SPPF(Fast)在于:重用了上一个maxpool的结果,第一次maxpool5,第二次在此基础上继续maxpool5则等同于maxpool9,第三次在此基础上继续maxpool5则等同于maxpool13,最后拼接原x,maxpool5结果,maxpool9结果,maxpool13结果,再用一个卷积调制通道与输入相同。

YOLOv5-6.0性能

YOLOv5结构讲解_第5张图片

以YOLOv5l对比5.0和6.0

YOLOv5结构讲解_第6张图片

 

YOLOv5s-5.0整体结构图

YOLOv5结构讲解_第7张图片

YOLOv5s-6.0整体结构图

YOLOv5结构讲解_第8张图片

 YOLOv5n-6.0整体结构图

YOLOv5结构讲解_第9张图片

 YOLOv5n-6.0整体结构图YOLOv5结构讲解_第10张图片

 与YOLOv5s-6.0不同在于每层通道数都变为其一半。

如有错误,望指正!

你可能感兴趣的:(深度学习)