不同框架的绘制神经网络结构可视化

seven_不是赛文

已于 2022-07-09 22:05:05 修改

阅读量447

点赞数

分类专栏：训练自己的网络文章标签：神经网络深度学习 keras

于 2022-06-26 22:09:05 首次发布

本文链接：https://blog.csdn.net/weixin_51015047/article/details/125475164

版权

训练自己的网络专栏收录该内容

7 篇文章 0 订阅

订阅专栏

一、keras

1，在keras中使用keras.summary()，或者plot_model()

eg：
import numpy as np
import matplotlib.pyplot as plt
from keras.utils.vis_utils import plot_mode
plot_model(model,to_file="model.png",show_shapes=True,show_layer_names=True,rankdir='TB')
plt.figure(figsize=(10,10))
img = plt.imread("model.png")
plt.imshow(img)
plt.axis('off')
plt.show()

在这里插入图片描述

二、PyTorch

输出网络结构：

MyConvNet = ConvNet()
 print(MyConvNet)
# 输出结果：

ConvNet(
   (conv1): Sequential(
     (0): Conv2d(1, 16, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))
     (1): ReLU()
     (2): AvgPool2d(kernel_size=2, stride=2, padding=0)
   )
   (conv2): Sequential(
     (0): Conv2d(16, 32, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))
     (1): ReLU()
     (2): MaxPool2d(kernel_size=2, stride=2, padding=0, dilation=1, ceil_mode=False)
   )
   (fc): Sequential(
     (0): Linear(in_features=1568, out_features=128, bias=True)
     (1): ReLU()
     (2): Linear(in_features=128, out_features=64, bias=True)
     (3): ReLU()
   )
   (out): Linear(in_features=64, out_features=10, bias=True)
）