2021 山东大学软件学院 软件工程应用与实践--YOLOV5项目代码分析——（6）

2021SC@SDUSC

再分析experiment.py文件，这个模块是yolov5的实验模块。

experiment.py

import numpy as np     # numpy矩阵操作模块
import torch           # PyTorch深度学习模块
import torch.nn as nn  # PYTorch模块函数库

from models.common import Conv, DWConv
from utils.google_utils import attempt_download

导入需要的模块

MixConv2d

MixConv2d 混合深度卷积来自于2019年发表的一篇论文：https://arxiv.org/abs/1907.09595. 源码：https://github.com/tensorflow/tpu/tree/master/models/official/mnasnet/mixnet. 混合深度卷积就是使用不同大小的卷积核对深度卷积的不同channel分组处理。也可以看作是 分组深度卷积 + Inception结构 的多种卷积核混用。

MixConv2d模块示意图：

 

class MixConv2d(nn.Module):
    """
    Mixed Depthwise Conv 混合深度卷积 就是使用不同大小的卷积核对深度卷积的不同channel分组