模型探索器
模型探索器提供了直观且层次化的模型图可视化。它将模型操作组织成嵌套层,使用户能够动态展开或折叠这些层。它还提供了一系列功能来促进模型探索和调试,包括突出显示输入和输出操作、在节点上叠加元数据、在交互式弹出窗口中显示层、执行搜索、显示相同的层、GPU加速图形渲染等。目前支持TFLite、TF、TFJS、MLIR和PyTorch(导出程序)模型格式,并为开发者提供了一个扩展框架,以便轻松添加对其他格式的支持。
要开始使用模型探索器,请运行:
$ pip install ai-edge-model-explorer
$ model-explorer
请查看我们的Wiki以获取更多详细信息:
我们邀请您参与模型探索器的研究调查。请在此处注册。
社区贡献的适配器
贡献
我们邀请您创建自定义适配器以添加对其他模型格式的支持。请参阅我们的开发适配器扩展指南。我们将乐意考虑接受拉取请求,以将您的适配器的GitHub仓库和PyPI包的链接添加到此README中。