为何说Netron是AI大模型可视化神器?

前有科技后进阶 2024-06-09 15:58:21

大家好,很高兴又见面了,我是"高级前端‬进阶‬",由我带着大家一起关注前端前沿、深入前端底层技术,大家一起进步,也欢迎大家关注、点赞、收藏、转发,您的支持是我不断创作的动力。

什么是 Netron

Visualizer for neural network, deep learning and machine learning models

Netron 是神经网络、深度学习和机器学习模型的查看器 (viewer),支持 ONNX、TensorFlow Lite、Core ML、Keras、Caffe、Darknet、MXNet、PaddlePaddle、ncnn、MNN 和 TensorFlow.js。

同时,Netron 为 PyTorch、TorchScript、TensorFlow、OpenVINO、RKNN、MediaPipe、ML.NET 和 scikit-learn 提供实验性支持。

目前 Netron 在 Github 通过 MIT 协议开源,有超过 26k 的 star、3k 的 fork、妥妥的前端顶级开源项目。

如何使用 Netron

以下是在不同平台上下载和使用 Netron 的示例:

macOS:下载 .dmg 文件或运行 brew install --cask netron,地址为:https://github.com/lutzroeder/netron/releases/latestLinux:下载 .AppImage 文件或运行 snap install netron,地址为:https://github.com/lutzroeder/netron/releases/latestWindows:下载 .exe 安装程序或运行 winget install -s winget netron,地址为:https://github.com/lutzroeder/netron/releases/latest浏览器:启动浏览器版本,地址为:https://netron.app/Python 服务器:运行 pip install netron 和 netron [FILE] 或 netron.start('[FILE]')

可以使用浏览器版本下载或打开以下示例模型文件,具体地址可以参考文末资料。

ONNX: squeezenetTensorFlow Lite: yamnetTensorFlow: chessbotKeras: mobilenetTorchScript: traced_online_pred_layerCore ML: exermoteDarknet: yolo

更多关于 Netron 的用法和示例可以参考文末,本文不再过多展开。

参考资料

https://github.com/lutzroeder/netron

https://netron.app/

https://www.youtube.com/watch?v=m9sfCvqH3Hw

https://www.youtube.com/watch?v=qh6oTF1Dj38

0 阅读:38

前有科技后进阶

简介:感谢大家的关注