仅用Chat模式,写了一个模块化显示神经网络的软件

使用 netron 查看网络结构时,发现大网络的结构展示效果不佳,于是用trae的chat功能开发了一款快速上手的模块化 ONNX 可视化工具(windows 版本)。

该工具支持查看模型结构,可将结构保存为 svg、png 格式图片,支持颜色切换,能查看模型基础信息(版本、参数量等),显示各 block 的 shape,还有放大缩小功能,方便查看细节。同时,还可以查看 conv 等网络层的数据分布情况,满足日常查看和分析模型的需求。

本人式C++和python选手,不太会其他语言,之前也没有任何开发软件的经验(全都看算法了),基本上下班有空的时候就会和chat模式对话,帮我写写功能(别问我为什么不用agent,因为当时不知道 :sob: ),一共耗时5个月,一周也就3-4个小时的开发时间(加班太严重)。顺便录了一个介绍视频

效果展示

这里导入Resnet50:

可以看到他不会直接把所有的网络结构都显示出来,而是分块显示,如果想看其中一个块的内容,继续点进去就好

如果这个模块就是一个基本模块,还可以查看信息,比如conv这个网络训好之后网络的参数分布情况:

1 个赞