您当前的位置: 首页 >  网络

耐心的小黑

暂无认证

  • 1浏览

    0关注

    323博文

    0收益

  • 0浏览

    0点赞

    0打赏

    0留言

私信
关注
热门博文

简单易用的神经网络模型可视化工具--netron

耐心的小黑 发布时间:2020-03-26 20:44:42 ,浏览量:1

       netron是一个神经网络模型可视化工具,下面对它做一下简单的介绍。

一、使用方式

  • windows系统下载.exe应用程序,Linux和macOS系统下载对应应的用程序即可,点击这里去下载。打开应用程序,直接添加模型即可查看网络结构。
  • 直接使用pip install netron安装netron包,然后使用以下程序去调用。
import netron
modelPath = "model.pb"
netron.start(modelPath)

此时它会自动跳转到浏览器显示模型结构。

二、目前支持的框架及其模型格式

Netron 支持的框架和对应文件如下:

框架 模型文件格式 ONNX (.onnx, .pb, .pbtxt) Keras (.h5, .keras)
关注
打赏
1640088279
查看更多评论
立即登录/注册

微信扫码登录

0.0376s