Tensorflow读取并输出已保存模型的权重数值方式
这篇文章是为了对网络模型的权重输出,可以用来转换成其他框架的模型。
importtensorflowastf fromtensorflow.pythonimportpywrap_tensorflow #首先,使用tensorflow自带的python打包库读取模型 model_reader=pywrap_tensorflow.NewCheckpointReader(r"model.ckpt") #然后,使reader变换成类似于dict形式的数据 var_dict=model_reader.get_variable_to_shape_map() #最后,循环打印输出 forkeyinvar_dict: print("variablename:",key) print(model_reader.get_tensor(key))
结果(其中一个权重的示例)
........ variablename:InceptionV3/Mixed_7c/Branch_3/Conv2d_0b_1x1/weights [[[[0.00013783-0.002514280.02235526...-0.014097020.00340105 -0.00752808] [0.01590012-0.00258413-0.00627338...-0.036004930.01220086 -0.01254225] [-0.02394262-0.00764508-0.00895328...-0.017314050.03568469 0.00918952] ... [-0.01865693-0.00358359-0.02342288...0.009358380.00367858 -0.00976252] [0.012976420.002234570.00652326...-0.00762609-0.0136022 -0.01129473] [-0.01395879-0.009202460.01061887...0.02369580.00041993 -0.01291134]]]] ......
以上这篇Tensorflow读取并输出已保存模型的权重数值方式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持毛票票。
声明:本文内容来源于网络,版权归原作者所有,内容由互联网用户自发贡献自行上传,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任。如果您发现有涉嫌版权的内容,欢迎发送邮件至:czq8825#qq.com(发邮件时,请将#更换为@)进行举报,并提供相关证据,一经查实,本站将立刻删除涉嫌侵权内容。