神经网络、深度学习和机器学习模型的可视化工具-netron

mtain 2023年12月20日 305次浏览

一、说明

Netron是一种用于神经网络、深度学习和机器学习模型的可视化工具,它可以为模型的架构生成具有描述性的可视化(descriptive visualization)。

支持的模型格式:
ONNX: squeezenet
TensorFlow Lite: yamnet
TensorFlow: chessbot
Keras: mobilenet
TorchScript: traced_online_pred_layer
Core ML: exermote
Darknet: yolo

Github:https://github.com/lutzroeder/netron

二、安装使用

在线版本

https://lutzroeder.github.io/netron/

操作系统

windows、mac等操作系统的可执行文件

python方式

# 安装
pip install netron

# 启动
netron

# 浏览器访问
http://localhost:8080/

三、使用

ch_PP-OCRv2_det_infer_opt.nb