代码
yan__csdn
这个作者很懒,什么都没留下…
展开
-
TensorFlow -4种交叉熵函数实现
https://blog.csdn.net/marsjhao/article/details/72630147转载 2018-09-23 11:15:51 · 597 阅读 · 0 评论 -
TensorFlow 抽取某一层特征
深度学习具有强大的特征表达能力。有时候我们训练好分类模型,并不想用来进行分类,而是用来提取特征用于其他任务,比如相似图片计算。接下来讲下如何使用TensorFlow提取特征。 1.必须在模型中命名好要提取的那一层,如下 self.h_pool_flat = tf.reshape(self.h_pool, [-1, num_filters_total], name='h_pool_fl...转载 2018-10-11 10:39:45 · 2965 阅读 · 0 评论 -
打印checkpoint参数
#断点续训练 ckpt = tf.train.get_checkpoint_state(FLAGS.train_dir) if ckpt and ckpt.model_checkpoint_path: print('restore success ') saver.restore(sess,ckpt...原创 2018-10-11 14:32:10 · 7019 阅读 · 0 评论 -
TensorFlow 加载模型前n-1层
path_checkpointa=r'E:\anacondaCode\ResNet_cifar-master181010(2000) - 副本\data\cifar10_tf\checkpoint_' ckpt = tf.train.get_checkpoint_state(path_checkpointa)#模型A的checkpoint if ckpt an...原创 2018-10-11 18:54:07 · 1545 阅读 · 0 评论