【可视化】Hiddenlayer可视化ResNeSt网络结构

  Hiddenlayer库包含一个build_graph()函数，可以非常方便地将深度学习网络进行可视化
1.AlexNet可视化
import torch
from torchvision.models.alexnet import AlexNet
import hiddenlayer as hl

model = AlexNet()
print(model)
hl_graph = hl.build_graph(model, torch.zeros([3, 3, 224, 224]))
hl_graph.theme = hl.graph.THEMES['blue'].copy()
hl_graph.save('./AlexNet.png', format='png')
print('Have done!')

AlexNet结构： 可视化：
2.ResNet18可视化
import torch

【可视化】Hiddenlayer可视化ResNeSt网络结构最先出现在Python成神之路。