从模型pt取出文件中的权重参数,绘制柱状图
- 一、详细步骤
-
-
- 第一步:读取pt变量中的文件content中,并查看content字典中键有以下两种方。
-
- 1.1 加载pt查看文件后content中的key
-
- 1.2 pt文件保存txt文件:成功!
- 1.3 过滤content中,取出model中含有weight关键字的key
- 获取model中某一key的shape
- 1.3 获取content中的(model)key对应的value
- 1.4 取出model中卷积层值
-
- 二、从模型pt取出文件中的权重参数,存储数据excel表格中-完整代码附上
-
-
- 结果图-成功!
-