Netron是一个用于可视化深度学习 络模型的工具软件,主体以JavaScript语言实现,源码在: https://github.com/lutzroeder/netron 。用户可以使用各系统平台的安装包进行安装之后使用,也可以用浏览器使用在线版本: Netron。
Netron支持几乎所有常见的深度学习框架所导出的模型: ONNX, TensorflowLite, Caffe, Keras, MNN, NCNN, MediaPipe等等。
实现效果如下图所示:
一.开发环境与基础依赖
- Windows 10 64bit: VSCode
- Python3: PyQt5-Qt5==5.15.2, PyQtWebEngine==5.15.4
- FlatWidget: PyQT图形界面设计二.扁平化无边框窗口
- Netron: https://github.com/lutzroeder/netron
二.软件集成
QWebEngine需以OpenGL Widget为容器进行创建,现在QT Designer中拖拽一个OpenGL Widget到主窗口中。
编译完成后,构造QWebEngineView:
只需调用此函数即可启用Netron。具体代码相机如下:
1~2行: 将netron的目录添加到python的环境中。
4~7行: 添加一个垂直布局器,并将QWebEngineView添加到其中。
9行: start为netron的API
file为模型文件的路径,data为模型数据,data与file都可以是空,都为空的情况下则打开的欢迎界面,browse指示是否自动打开本地浏览器查看,该函数的返回值为本地http服务器的地址与端口:(address, port)。
11行,通过QWebEngine加载serve.start启动的http服务。
文章知识点与官方知识档案匹配,可进一步学习相关知识Python入门技能树首页概览210545 人正在系统学习中
声明:本站部分文章及图片源自用户投稿,如本站任何资料有侵权请您尽早请联系jinwei@zod.com.cn进行处理,非常感谢!