AI学习
崔林繁
这个作者很懒,什么都没留下…
展开
-
如何从ckpt中得到层名和参数量
import tensorflow as tfsaver = tf.train.import_meta_graph('xxx.meta')variables = tf.trainable_variables()total_parameters = 0for variable in variables: shape = variable.get_shape() variable_parameters = 1 for dim in shape: variable_原创 2020-05-21 14:46:12 · 376 阅读 · 0 评论 -
pth文件数据修改
在调用github上下载的代码后model.load_state_dict(a.pth)编译不过的情况使用了 model.load_state_dict(a.pth,strict = False)依旧不能解决后来通过分析模型发现,确实是某些层的参数量不匹配,但是这些层的参数其实通过复制成需要的大小就可以解决问题,所以有了以下的方法1. 分析pth的结构import torchfy=to...原创 2020-04-24 10:18:34 · 4386 阅读 · 0 评论 -
pytorch AI 网络格式分析方法
1.分析 *.pth的网络结构import torchfy=torch.load(路径)for i in fy.keys() print(i+' '+str(list(fy[i].size())))2.分析自定义网络结构import torchsummarysummary(model,input_size=(3,224,224))产生错误input 和weight 的...原创 2020-04-24 10:05:09 · 107 阅读 · 0 评论 -
论文Background Matting: The World is Your Green Screen学习
使用 a deep matting network 为给定的input frame 提取foreground color 和 alpha。使用了一个novel 技术“context-switching block”,这个技术可以合并有选择性的合并不同的输入线索。...原创 2020-04-08 13:33:27 · 2664 阅读 · 0 评论