yolov5网络结构代码解读

发布时间：2023-04-21 文章分类：电脑百科 投稿人：王小丽 字号： 默认 | 大 | 超大 打印

文章目录

  • 前言
  • 1.项目介绍
  • 2.yolov5的网络结构
    • 1.yolov5s的配置文件
    • 2.网络模型的初始化和训练过程
    • 3.backbone
    • 4.head
      • 1.三层预测
      • 2.4层预测
    • 5.detect
  • 补充个更直观点的图
  • 总结

前言

yolov5已经很成熟了，作为一个拥有发展系列的检测器，它拥有足够的精度和满足现实中实时性要求，所以许多项目和比赛都能用的上，自己也拿来参加过比赛。
本博客的讲解代码来源：https://github.com/ultralytics/yolov5

1.项目介绍

YOLOv5针对不同大小的输入和网络深度宽度，主要分成了（n, s, m, l, x）和（n6, s6, m6, l6, x6），这些都在yolov5的项目代码的配置文件中有对应。其中随着版本的更新，里面也多了好多其他模块。这里，我主要用的是v6.0版本。
yolov5网络结构代码解读

2.yolov5的网络结构

这里主要结合代码介绍下yolov5sv6.0的网络结构部分，其他大小的框架都差不多。
如下图所示，这里给出了我参考一些yolov5图根据6.0代码所画的yolov5l网络

结构图：
yolov5网络结构代码解读

1.yolov5s的配置文件

相关参数

# YOLOv5  by Ultralytics, GPL-3.0 license
# Parameters
nc: 4  # number of classes，检测的类别
depth_multiple: 0.33  # model depth multiple, 决定下面的 number:n
width_multiple: 0.50  # layer channel multiple，解决网络的深度和宽度
anchors:
  - [10,13, 16,30, 33,23]  # P3/8		 # anchor尺寸设置
  - [30,61, 62,45, 59,119]  # P4/16
  - [116,90, 156,198, 373,326]  # P5/32	 # 可以自己手动设置，也可以自动聚类

backbone
里面的-1表示自身的特征层的位置

# YOLOv5 v6.0 backbone
backbone:
  # [from, number, module, args[C,K,S,P]] P:会根据公式自动推导，配置文件也看不太出。
#分别对应：[输入位置，叠加层数，使用模块名称，[输出通道数，卷积核大小，步距，padding]]
  [[-1, 1, Conv, [64, 6, 2, 2]],  # 0-P1/2
   [-1, 1, Conv, [128, 3, 2]],  # 1-P2/4
   [-1, 3, C3, [128]],
   [-1, 1, Conv, [256, 3, 2]],  # 3-P3/8
   [-1, 6, C3, [256]],
   [-1, 1, Conv, [512, 3, 2]],  # 5-P4/16
   [-1, 9, C3, [512]],
   [-1, 1, Conv, [1024, 3, 2]],  # 7-P5/32
   [-1, 3, C3, [1024]],
   [-1, 1, SPPF, [1024, 5]],  # 9，总共
  ]

head
里面的[-1, 6]表示自身特征层和第6个位置上的特征层。如[[-1, 6], 1, Concat, [1]]，表示和backbone里显示的P4那层特征层相cat。

# YOLOv5 v6.0 head
head:
  [[-1, 1, Conv, [512, 1, 1]],
   [-1, 1, nn.Upsample, [None, 2, 'nearest']],
   [[-1, 6], 1, Concat, [1]],  # cat backbone P4
   [-1, 3, C3, [512, False]],  # 13
   [-1, 1, Conv, [256, 1, 1]],
   [-1, 1, nn.Upsample, [None, 2, 'nearest']],
   [[-1, 4], 1, Concat, [1]],  # cat backbone P3
   [-1, 3, C3, [256, False]],  # 17 (P3/8-small)
   [-1, 1, Conv, [256, 3, 2]],
   [[-1, 14], 1, Concat, [1]],  # cat head P4
   [-1, 3, C3, [512, False]],  # 20 (P4/16-medium)
   [-1, 1, Conv, [512, 3, 2]],
   [[-1, 10], 1, Concat, [1]],  # cat head P5
   [-1, 3, C3, [1024, False]],  # 23 (P5/32-large)
   [[17, 20, 23], 1, Detect, [nc, anchors]],  # Detect(P3, P4, P5)
  ]

2.网络模型的初始化和训练过程

根据配置文件，初始化网络模型：
代码路径：yolov5-master/models/yolo.py

def parse_model(d, ch):  # model_dict, input_channels(3)
    LOGGER.info(f"\n{'':>3}{'from':>18}{'n':>3}{'params':>10}{'module':<40}{'arguments':<30}")
    anchors, nc, gd, gw = d['anchors'], d['nc'], d['depth_multiple'], d['width_multiple']
    na = (len(anchors[0]) // 2) if isinstance(anchors, list) else anchors  # number of anchors,每一个predict head上的anchor数 = 3
    no = na * (nc + 5)  # number of outputs = anchors * (classes + 5)
    #  layers: 保存每一层的层结构 save: 记录下所有层结构中from中不是-1的层结构序号 c2: 保存当前层的输出channel
    layers, save, c2 = [], [], ch[-1]  # layers, savelist, ch out
    # from(当前层输入来自哪些层), number(当前层次数 初定), module(当前层类别), args(当前层类参数 初定)
    for i, (f, n, m, args) in enumerate(d['backbone'] + d['head']):  # from, number, module, args
        m = eval(m) if isinstance(m, str) else m  # eval strings
        for j, a in enumerate(args):
            try:
                args[j] = eval(a) if isinstance(a, str) else a  # eval strings
            except NameError:
                pass
        n = n_ = max(round(n * gd), 1) if n > 1 else n  # depth gain:控制深度  如v5s: n*0.33 n: 当前模块的次数(间接控制深度)
        if m in (Conv, GhostConv, Bottleneck, GhostBottleneck, SPP, SPPF, DWConv, MixConv2d, Focus, CrossConv,
                 BottleneckCSP, C3, C3TR, C3SPP, C3Ghost, nn.ConvTranspose2d, DWConvTranspose2d, C3x,ACConv,CAM_Module):
            c1, c2 = ch[f], args[0]  # c1: 当前层的输入的channel数  c2:当前层的输出的channel数(初定)  ch:记录着所有层的输出channel
            if c2 != no:  # if not output,最后一层
                c2 = make_divisible(c2 * gw, 8)  #  通道数调整(64*0.5,8)
            args = [c1, c2, *args[1:]]  # 
            if m in [BottleneckCSP, C3, C3TR, C3Ghost, C3x]:
                args.insert(2, n)  # number of repeats
                n = 1
        elif m is nn.BatchNorm2d:
            args = [ch[f]]
        elif m is Concat:
            c2 = sum(ch[x] for x in f)
        elif m is Detect:
            args.append([ch[x] for x in f])
            if isinstance(args[1], int):  # number of anchors
                args[1] = [list(range(args[1] * 2))] * len(f)
        elif m is Contract:
            c2 = ch[f] * args[0] ** 2
        elif m is Expand:
            c2 = ch[f] // args[0] ** 2
        else:
            c2 = ch[f]

顺序执行网络的训练过程：

		m_ = nn.Sequential(*(m(*args) for _ in range(n))) if n > 1 else m(*args)  # module
		        t = str(m)[8:-2].replace('__main__.', '')  # module type
		        np = sum(x.numel() for x in m_.parameters())  # number params:计算参数量
		        m_.i, m_.f, m_.type, m_.np = i, f, t, np  # attach index, 'from' index, type, number params
		        LOGGER.info(f'{i:>3}{str(f):>18}{n_:>3}{np:10.0f}{t:<40}{str(args):<30}')  # print
		        save.extend(x % i for x in ([f] if isinstance(f, int) else f) if x != -1)  # append to savelist
		        layers.append(m_)
		        if i == 0:
		            ch = []
		        ch.append(c2)
		    return nn.Sequential(*layers), sorted(save)

3.backbone

backbone：特征提取网络。由yolov3的Darknet53变为yolov4的CSPDarknet53，yolov5里较小改动。
yolov5网络结构代码解读

根据配置文件解析整个backbone的结构：
yolov5网络结构代码解读

网络结构代码路径：yolov5-master/models/common.py

class Conv(nn.Module):
    # Standard convolution
    def __init__(self, c1, c2, k=1, s=1, p=None, g=1, act=True):  # ch_in, ch_out, kernel, stride, padding, groups
        super().__init__()
        self.conv = nn.Conv2d(c1, c2, k, s, autopad(k, p), groups=g, bias=False)
        self.bn = nn.BatchNorm2d(c2)
        self.act = nn.SiLU() if act is True else (act if isinstance(act, nn.Module) else nn.Identity())
    def forward(self, x):
        return self.act(self.bn(self.conv(x)))
    def forward_fuse(self, x):
        return self.act(self.conv(x))

CSPNet的优点：
1.Strengthening learning ability of a CNN:现有的CNN在轻量化后，其精度大大降低，因此希望加强CNN的学习能力，使其在轻量化的同时保持足够的准确性。
2.Removing computational bottlenecks:希望能够均匀分配CNN中各层的计算量，这样可以有效提升各计算单元的利用率，从而减少不必要的能耗。
3.Reducing memory costs:在减少内存使用方面，采用cross-channel pooling，在特征金字塔生成过程中对特征图进行压缩。

yolov5网络结构代码解读

class C3(nn.Module):
    # CSP Bottleneck with 3 convolutions
    def __init__(self, c1, c2, n=1, shortcut=True, g=1, e=0.5):  # ch_in, ch_out, number, shortcut, groups, expansion
        super().__init__()
        c_ = int(c2 * e)  # hidden channels，1/2
        self.cv1 = Conv(c1, c_, 1, 1)
        self.cv2 = Conv(c1, c_, 1, 1)
        self.cv3 = Conv(2 * c_, c2, 1)  # optional act=FReLU(c2)
        self.m = nn.Sequential(*(Bottleneck(c_, c_, shortcut, g, e=1.0) for _ in range(n)))
    def forward(self, x):
        return self.cv3(torch.cat((self.m(self.cv1(x)), self.cv2(x)), 1))

class SPPF(nn.Module):
    # Spatial Pyramid Pooling - Fast (SPPF) layer for YOLOv5 by Glenn Jocher
    def __init__(self, c1, c2, k=5):  # equivalent to SPP(k=(5, 9, 13))
        super().__init__()
        c_ = c1 // 2  # hidden channels
        self.cv1 = Conv(c1, c_, 1, 1)
        self.cv2 = Conv(c_ * 4, c2, 1, 1)
        self.m = nn.MaxPool2d(kernel_size=k, stride=1, padding=k // 2)
    def forward(self, x):
        x = self.cv1(x)
        with warnings.catch_warnings():
            warnings.simplefilter('ignore')  # suppress torch 1.9.0 max_pool2d() warning
            y1 = self.m(x)
            y2 = self.m(y1)
            return self.cv2(torch.cat((x, y1, y2, self.m(y2)), 1))

4.head

1.三层预测

head:yolov5配置文件里写的head，其实就是对应通用检测模块里的neck。就是为了更好的检测不同尺度目标大小设计的特征金字塔结构。结构如下图所示：
yolov5网络结构代码解读

2.4层预测

Yolov5l6,m6,n6,x6都是用四层来预测输出的。

# yolov5l6：
# YOLOv5 v6.0 backbone
backbone:
  # [from, number, module, args]
  [[-1, 1, Conv, [64, 6, 2, 2]],  # 0-P1/2
   [-1, 1, Conv, [128, 3, 2]],  # 1-P2/4
   [-1, 3, C3, [128]],
   [-1, 1, Conv, [256, 3, 2]],  # 3-P3/8
   [-1, 6, C3, [256]],
   [-1, 1, Conv, [512, 3, 2]],  # 5-P4/16
   [-1, 9, C3, [512]],
   [-1, 1, Conv, [768, 3, 2]],  # 7-P5/32
   [-1, 3, C3, [768]],
   [-1, 1, Conv, [1024, 3, 2]],  # 9-P6/64
   [-1, 3, C3, [1024]],
   [-1, 1, SPPF, [1024, 5]],  # 11
  ]
# YOLOv5 v6.0 head
head:
  [[-1, 1, Conv, [768, 1, 1]],      # [-1, 1, Conv, [768, 1, 1]],[-1, 1, CAM_Module, [768, 1, 1]],
   [-1, 1, nn.Upsample, [None, 2, 'nearest']],
   [[-1, 8], 1, Concat, [1]],  # cat backbone P5
   [-1, 3, C3, [768, False]],  # 15
   [-1, 1, Conv, [512, 1, 1]],
   [-1, 1, nn.Upsample, [None, 2, 'nearest']],
   [[-1, 6], 1, Concat, [1]],  # cat backbone P4
   [-1, 3, C3, [512, False]],  # 19
   [-1, 1, Conv, [256, 1, 1]],
   [-1, 1, nn.Upsample, [None, 2, 'nearest']],
   [[-1, 4], 1, Concat, [1]],  # cat backbone P3
   [-1, 3, C3, [256, False]],  # 23 (P3/8-small)
   [-1, 1, Conv, [256, 3, 2]],
   [[-1, 20], 1, Concat, [1]],  # cat head P4
   [-1, 3, C3, [512, False]],  # 26 (P4/16-medium)
   [-1, 1, Conv, [512, 3, 2]],
   [[-1, 16], 1, Concat, [1]],  # cat head P5
   [-1, 3, C3, [768, False]],  # 29 (P5/32-large)
   [-1, 1, Conv, [768, 3, 2]],
   [[-1, 12], 1, Concat, [1]],  # cat head P6
   [-1, 3, C3, [1024, False]],  # 32 (P6/64-xlarge)
   [[23, 26, 29, 32], 1, Detect, [nc, anchors]],  # Detect(P3, P4, P5, P6)
  ]

通过在backbone里多加一层768通道的特征层，会使最顶层的分辨率继续按1/4减小，长宽减小一半。
然后经过自上而下和自下而上的特征层匹配coancat融合，会多出一个P6预测层。多用一个预测层会有什么好处呢：能够检测更大的目标物体，提取的目标特征语义信息更丰富，自上而下的传递的语义信对各个层更好。

5.detect

通过卷积预测输出相应通道数的特征层用于分类和回归。
yolov5网络结构代码解读
c = (5+num_cls)x3:(四个坐标偏移值+1个置信度+预测的类别数)x每个像素所给3个anchors。
Conv:这里的就是普通的1x1卷积。

补充个更直观点的图

yolov5网络结构代码解读

总结

这里对Yolov5的网络结构部分进行了总结，后续有时间，再对其他部分做总结。

标签 , ,