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)
省了模型上传的步骤。
懒是进步的动力。
网友评论