模型可视化工具Netron(支持tf, caffe, keras,onnx等多种框架)
netron工具的官网下载请参考:https://github.com/lutzroeder/Netron
但本人尝试下载windows版本,GitHub下载失败,所以搜索汇总以下两种解决方案:
- 网页版的Netron
输入如下链接:https://lutzroeder.github.io/netron/ 即可弹出网页版,直接单击网页中的Open Model按钮,选择需要可视化的文件即可打开可视化网络。 - 话说是python版的Netron
第一步:安装Netron
pip install netron
第二步:代码实现
import netron
netron.start(R"E:\tensorflow\fft_bp_v0.3\model_information\bpmodel.h5")
运行该python代码即会自动打开默认的浏览器,然后在浏览器中输入:http://localhost:8080 即可得到同方法1中相同的效果。