分享一个模型结构可视化工具
对于pytorch、caffe、keras、tensorflow、mxnet等搭建的网络结构,往往不能很直观的看出来,那么今天推荐一个能非常直观的能看到搭建的模型的整体模型结构的工具软件——Netron。
Netron 支持Windows、Linux、Mac系统。
Linux安装直接到github上面拉下来,setup就行了。
框架对应的文件类型
框架 | 文件 |
---|---|
onnx | .onnx、.pb |
keras | .h5、.keras |
tensorflow | .pb、.meta |
mxnet | .model |
caffe | .caffemodel |
对于现在最常用的深度学习框架pytorch来说,首先需要将.pth文件转换为.onnx文件,再用Netron打开查看模型结构可视化图。
# 模型保存onnx文件
torch.onnx._export(model, input, "1.onnx", verbose=True, opset_version=11)