分类：ckpt文件

2020
09-29

tensorflow实现从.ckpt文件中读取任意变量

思路有些混乱，希望大家能理解我的意思。看了fasterrcnn的tensorflow代码，关于fix_variables的作用我不是很明白，所以写了以下代码，读取了预训练模型vgg16得fc6和fc7的参数，以及fasterrcnn中heat_to_tail中的fc6和fc7，将它们做了对比，发现结果不一样，说明vgg16的fc6和fc7只是初始化了fasterrcnn中heat_to_tail中的fc6和fc7，之后后者被训练。具体读取任意变量的代码如下：importtensorflowastfimportnumpyasnpfrom... 继续阅读 >