模型结构可视化

如何对caffe的网络模型进行可视化?

一、安装netron

1.windows版本:

    下载可执行文件或在终端运行: winget install netron

2.Linux版本:

   运行:snap install netron

   启动:netron

 

Netron不仅支持Caffe(caffemodel),还支持pytorch(pth),ONNX(.onnx, .pb)

 

posted @ 2021-04-20 23:54  jimchen1218  阅读(229)  评论(0)    收藏  举报