Yolov5 网络构建代码（2）- Model

在该目录下存放着yolo.py文件，里面的代码是关于网络构建相关的。

里面其实就写了两个class，一个是Detect，一个是Model

Model

首先我们看先看__init__初始化方法

 def __init__(self, cfg='yolov5s.yaml', ch=3, nc=None, anchors=None):  # model, input channels, number of classes
        super().__init__()
        if isinstance(cfg, dict):
            self.yaml = cfg  # model dict
        else:  # is *.yaml
            import yaml  # for torch hub
            self.yaml_file = Path(cfg).name
            with open(cfg, errors='ignore') as f:
                self.yaml = yaml.safe_load(f)  # model dict

        # Define model
        ch = self.yaml['ch'] = self.yaml.get('ch', ch)  # input channels
        if nc and nc != self.yaml['nc']:
            LOGGER.info(f"Overriding model.yaml nc={self.yaml['nc']} with nc={nc}")
            self.yaml['nc'] = nc  # override yaml value
        if anchors:
            LOGGER.info(f'Overriding model.yaml anchors with anchors={anchors}')
            self.yaml['anchors'] = round(anchors)  # override yaml value
        self.model, self.save = parse_model(deepcopy(self.yaml), ch=[ch])  # model, savelist
        self.names = [str(i) for i in range(self.yaml['nc'])]  # default names
        self.inplace = self.yaml.get('inplace', True)

        # Build strides, anchors
        m = self.model[-1]  # Detect()
        if isinstance(m, Detect):
            s = 256  # 2x min stride
            m.inplace = self.inplace
            m.stride = torch.tensor([s / x.shape[-2] for x in self.forward(torch.zeros(1, ch, s, s))])  # forward
            m.anchors /= m.stride.view(-1, 1, 1)
            check_anchor_order(m)
            self.stride = m.stride
            self._initialize_biases()  # only run once

        # Init weights, biases
        initialize_weights(self)
        self.info()
        LOGGER.info('')

一开始是接受的了4个数据  cfg='yolov5s.yaml', ch=3, nc=None, anchors=None

cfg：网络架构配置文件

ch：输入通道数，默认为3

nc：总类别数

anchors：先验框参数。

然后是判断输入的cfg(网络架构文件)是否为字典。一般都不是字典，直接进入else，打开yaml文件，转换成字典格式：

if isinstance(cfg, dict):
    self.yaml = cfg  # model dict
else:  # is *.yaml
    import yaml  # for torch hub
    self.yaml_