Tensorflow vgg_16.ckpt预训练权重读取
import tensorflow as tf
import os
model_dir = './vgg_weights_read/checkpoint'
model_name = 'vgg_16.ckpt'
pretrain_path = os.path.join(model_dir, model_name)
#先定义变量
var = tf.get_variable(name='vgg_1...
原创
2018-05-02 14:54:52 ·
6114 阅读 ·
0 评论