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