如何通过yaml配置文件在Pytorch中详细构建卷积神经网络？

本文共计2662个文字，预计阅读时间需要11分钟。

目录+定义卷积块+定义一个Bottleneck+传入参数+将参数传入模型+完整的代码+大多数卷积神经网络都是直接通过一个Model类来定义的，这样编写的代码实际上是比较易懂的，特别是在魔改网络时。

目录

• 定义卷积块
• 定义一个Bottleneck
• 传入参数
• 将参数传入模型
• 完整的代码

大多数卷积神经网络都是直接通过写一个Model类来定义的，这样写的代码其实是比较好懂的，特别是在魔改网络的时候也很方便。然后也有一些会通过cfg配置文件进行模型的定义。在yolov5中可以看到是通过yaml文件进行网络的定义，可能很多人也用过，如果自己去写一个yaml文件，自己能不能定义出来呢？很多人不知道是如何具体通过yaml文件将里面的参数传入自己定义的网络中，这也就给自己修改网络带来了不便。这篇文章将仿照yolov5的方式，利用yaml定义一个自己的网络。