本地使用netron

2023-11-22  本文已影响0人  DayDayUp_hhxx

1.安装netron

pip install netron

2.在linux的terminal中使用以下命令,然后打开浏览器查看。

netron   
netron   --host 0.0.0.0  --port 6780
netron "model_path"   --host 0.0.0.0  --port 6780

3.在jupyterlab中使用

github有人推荐的方法,jupyterlab部署在本地的可以用

import IPython

import netron
port = 6006
model_path = # <Your favorite model>
netron.start(model_path, 6006, browse=False)

IPython.display.IFrame(f"http://localhost:{port}", width=1000, height=1000)

jupyterlab部署在服务器上,IPython.display可能会拒绝连接,可以使用下面的方法:
在jupyter中执行以下命令,然后浏览器可以查看。

import netron
netron.start(“model的绝对路径”,('0.0.0.0',6780),browse=False)

省了模型上传的步骤。
懒是进步的动力。

上一篇 下一篇

猜你喜欢

热点阅读