Netron 是一个开源的深度学习可视化工具,它可以让你轻松地查看和理解深度学习模型的结构和权重。Netron 提供了多种格式模型的导入支持,包括 ONNX、TensorFlow、PyTorch 等。

特点

  • 多格式支持:Netron 支持 ONNX、TensorFlow、PyTorch 等多种深度学习模型的导入。
  • 交互式界面:Netron 提供了交互式界面,可以方便地查看模型的结构和权重。
  • 可扩展性:Netron 允许用户自定义模型查看器,扩展其功能。

使用方法

  1. 下载 Netron:Netron 下载链接
  2. 打开 Netron:双击下载的 Netron.exe 文件。
  3. 导入模型:点击菜单栏的 "File" -> "Open" 选择你的模型文件。

图片展示

这里有一个使用 Netron 查看模型的例子:

模型结构展示

相关链接