首页 PaddleCV 帖子详情
打印模型结构的问题
收藏
快速回复
PaddleCV 问答目标检测 2128 3
打印模型结构的问题
收藏
快速回复
PaddleCV 问答目标检测 2128 3

(1)请问一下,要查看ppcls或ppdet保存的模型结构,应该用什么指令或者操作呢?

(2)像下面这种是可以直接通过paddle.summary()进行打印模型结构,那如何打印上面自行设计的通过PaddleClas或PaddleDetection训练生成的网络结构?该如何加载训练模型

import paddle
model = paddle.vision.models.resnet50(pretrained=True)
paddle.summary(model, (64, 3, 224, 224))
0
收藏
回复
全部评论(3)
时间顺序
飞桨_PaddleGAN
#2 回复于2021-09

可视化网络结构可以使用飞桨可视化分析工具VisualDL:https://github.com/PaddlePaddle/VisualDL 

其中有一个graph功能,可以直接拖拽上传pdmodel文件即可直接查看,AIStudio也有这个可视化接口,可以直接点击选择对应文件进行可视化,不过近期在修复中,预计9.23日重新上线AIStudio,您可以在本地安装使用(pip install visualdl --upgrade),启动命令为 visualdl --model ./log/ppcls.pdmodel (--model后为模型结构文件所在路径)即可可视化

具体使用可见:https://github.com/PaddlePaddle/VisualDL/blob/develop/docs/components/README_CN.md#Graph--%E7%BD%91%E7%BB%9C%E7%BB%93%E6%9E%84%E7%BB%84%E4%BB%B6

0
回复
李长安
#3 回复于2021-12

优秀

0
回复
L
LZY26423
#4 回复于2022-07

你好,你查看成功了吗

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户