大道至简！如何高效率的画出深度学习网络图

文章预览

在看论文或者调试代码的过程，我们总希望能够对模型的网络图了然于心。 今天介绍一款非常适合画模型网络图的软件——飞书，当然你也可以用PPT或者draw.io绘制模型的网络图，其他乱七八糟的绘制模型的软件建议没必要去学了。 我们在绘制模型网络图前，需要了解模型的网络结构，并将网络结构划分多个公用的模块，然后我们就可以轻松绘制模型网络图了。 我们以最经典的resnet网络为例。 resnet论文的网络图： 根据上述图的网络图，我们模块化了网络图： 卷积模块 image-20240829224852462 残差单元模块1 ，当输入和输出的通道相同时，不需要通过1*1的卷积网络进行通道映射。 image-20240829225112465 残差单元模块2 ，当输入和输出的通道不相同时，需要通过1*1的卷积网络进行通道映射。 image-20240829225257881 残差单元模块3 ，当输入和输出的通道不相同且包含 ………………………………