神经网络模型可视化工具 Netron | 每日工具推荐

今天要给大家介绍的是神经网络模型可视化工具 Netron,Netron 支持 ONNX, TensorFlow Lite, Keras, Caffe, Darknet, ncnn, MNN, PaddlePaddle, Core ML, MXNet, RKNN, MindSpore Lite, TNN, Barracuda, Tengine, TensorFlow.js, Caffe2 and UFF。同时实验性支持 PyTorch, TensorFlow, TorchScript, OpenVINO, Torch, Vitis AI, Arm NN, BigDL, Chainer, CNTK, Deeplearning4j, MediaPipe, ML.NET and scikit-learn。好吧,罗列了那么一大堆,不用我说,大家也都知道它的强大了。。。。。

这里再重复说一下吧,其实 C 站内有很多的文章可以参考,比如下面搜索 Netron 就可以出来很多文章,甚至有些文章阅读量都有好几万,已经很不错了,反正就是和大家多说一句,不知道你们平时用不用这个方法。

神经网络模型可视化工具 Netron | 每日工具推荐_第1张图片

接下来我们看下安装,Netron 的话,支持 macOS、Linux、Windows 以及还有在线的网页版可供快捷使用。首先我们需要下载 exe 安装包,因为我演示的电脑是 Windows 系统,下载如下图所示的安装包。

神经网络模型可视化工具 Netron | 每日工具推荐_第2张图片

下载过来之后,我们可以双击 exe 程序,下面展示安装完之后的结果图:

神经网络模型可视化工具 Netron | 每日工具推荐_第3张图片

安装完之后,我们需要一些模型来做演示,看一下官网,我们可以看到有好多种示例模型可供使用,为了简单起见,我只挑选第一个 ONNX 来给大家演示下结果。

神经网络模型可视化工具 Netron | 每日工具推荐_第4张图片

首先我们需要下载 squeezenet 的 ONNX 格式模型,然后打开它,结果如下图所示:

神经网络模型可视化工具 Netron | 每日工具推荐_第5张图片

这篇虽然简单了一些些,但是这个工具确实值得推荐,所以也就写了。

你可能感兴趣的:(神经网络与深度学习,神经网络,深度学习,tensorflow)