Skip to content

visual

Jump to bottom
wangzhaode edited this page Feb 16, 2023 · 1 revision

可视化工具

可视化的效果： 可视化效果

在详细调研了市面上比较主流的可视化工具后，Netron是一款受众面较多、兼容多款模型的可视化模型，同时它还具有跨平台支持、Python模块支持的能力。因此，在研究了一番它的设计和架构并考虑后续MNN自身的演进，我们决定官方维护MNN模型的可视化能力并将其作为Pull Request合并，大家可以放心使用啦。

功能列表

- 支持加载`.mnn`模型 。
- 支持将可视化的图导出成图片保存。
- 支持拓扑结构的展示、`Operator`/`Input`/`Output`的内容展示。
- 支持结构化的`weight`，`scale`，`bias`等数据的展示，**并支持将此类数据持久化保存**。

使用方式（Release版本）

  • 下载地址
  • macOS: 下载 .dmg文件 或者 brew cask install netron
  • Linux: 下载 .AppImage或者.deb文件.
  • Windows: 下载.exe文件.
  • Python：pip install netron

使用开发版本

  • 对仓库地址：https://github.com/lutzroeder/netron，进行clone。始终使用master分支。
  • cd [your_clone_path]/netron
  • 安装npm，确保npm版本大于6.0.0
  • npm install

使用JavaScript调试

npx electron ./（如果这步失败，单独npm install -g npx）

使用Python调试

python3 setup.py build
export PYTHONPATH=build/lib:${PYTHONPATH}
python3 -c "import netron; netron.start(None)"

遗留问题

加载超大模型可能渲染失败（几千个节点）

Clone this wiki locally