DIY深度学习模型结构可视化工具

这个工具用于将自定义ONNX模型（及其他深度学习模型）的计算图可视化。它能够将模型中的每个节点、张量信息和连接关系以详细表格形式展示，帮助开发者直观理解复杂的神经网络结构。在工作中，当需要将自定义计算图可视化时，引擎开发程序员通常缺乏GUI设计能力，外包给前端程序员又耗时耗力。这个方案借助Graphviz和Python快速生成可视化图表，解决了项目紧急需求。生成的可视化结果可保存为SVG格式，这种矢量图形支持无级缩放，能够清晰查看连接细节。在浏览器中打开SVG文件后，可以直接按操作名或张量名搜索定位节点，极大提升了调试效率。该工具特别适用于模型优化、性能分析和架构理解场景，让开发者能够快速掌握计算图的数据流向和节点关系，无需依赖复杂的前端开发。