google-ai-edge/model-explorer
一款把深度学习模型内部结构画出来看的工具
Model Explorer 把深度学习模型的运算图以分层结构可视化出来。用户可以展开或折叠各层,高亮输入输出运算,并在节点上叠加显示元数据。它支持 TFLite、TF、TFJS、MLIR 和 PyTorch(Exported Program)格式,并提供扩展框架以支持更多格式。
它做什么
- 把模型运算组织成可展开折叠的嵌套层级,方便探索图结构
- 提供输入输出运算高亮、节点元数据叠加显示、交互式弹窗查看层、搜索、识别相同层、GPU加速渲染等功能
- 执行 pip install ai-edge-model-explorer 后运行 model-explorer 命令即可使用,也可以在 Hugging Face 上传模型直接体验
- 核心可视化组件也以 ai-edge-model-explorer-visualizer 这个 npm 包发布
- 社区已经为 ONNX、VGF、TOSA 格式开发了独立的适配器仓库
为什么重要
模型结构越来越复杂时,单靠阅读代码很难搞清楚各个运算是怎么连接的,这个工具让人可以直观地查看结构并进行调试。它的适配器扩展框架也让开发者可以自行添加对 TFLite、PyTorch 之外格式的支持。
本仓库术语
- TFLite · 用于在移动和嵌入式设备上运行的轻量级 TensorFlow 模型格式
- MLIR · 多个编译器共用的一种中间表示格式
- Exported Program · PyTorch 模型导出后得到的图结构表示
- 适配器(Adapter) · 把特定模型格式转换成该工具可读取形式的扩展模块
仓库简介(英文)
A modern model graph visualizer and debugger
在 GitHub 打开热门仓库
- vorssaint/vorssaint-utils一个免费菜单栏应用,顶替十几款付费Mac工具
- Alishahryar1/free-claude-code一个本地代理,把编程AI助手接到49家免费或低价模型服务商,不再只能用一家付费服务
- freestylefly/awesome-gpt-image-2把532个GPT-Image2案例拆解成可复用模板的提示词工程库
- block/buzz一个让人类和AI智能体在同一空间里聊天、写代码、做审查的开源协作平台
- NousResearch/hermes-agentNous Research推出的Hermes是一款越用越聪明的AI智能体
- virgiliojr94/book-to-skill把技术类书籍PDF变成AI编程助手随时可查阅的技能包
- VoltAgent/awesome-agent-skills把上千份让AI编程助手秒变专家的操作手册集中放在一个地方
- anthropics/claude-plugins-community一个把大家做的Claude插件集中起来供安装的仓库