利用netron查看pb文件的结构信息

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接: https://blog.csdn.net/weixin_38145317/article/details/98869458
import netron
modelPath = "/home/jerry/PY_project_wang/PY_process_data/process_images/model/person_head/resnet50_freeze20.pb"#输入pb文件的路径即可
netron.start(modelPath)#会打开一个浏览器,显示模型的结构信息

猜你喜欢

转载自blog.csdn.net/weixin_38145317/article/details/98869458