代码
yan__csdn
这个作者很懒,什么都没留下…
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
TensorFlow -4种交叉熵函数实现
https://blog.youkuaiyun.com/marsjhao/article/details/72630147转载 2018-09-23 11:15:51 · 626 阅读 · 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 · 3015 阅读 · 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 · 7161 阅读 · 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 · 1598 阅读 · 0 评论
分享