
tensorflow笔记
CHNguoshiwushuang
这个作者很懒,什么都没留下…
展开
-
使用tensorflow实现卷积的几种操作
这里简要总结一下tensorflow实现卷积操作的几种写法,定义方式有很多。例如使用一个9x9的卷积核将一个三通道图像转换成64通道的feature,遇到过的有以下一些: 直接定义就写 这种写法简单粗暴,一边定义卷积核、偏置和卷及操作一边写,优点是操作灵活,但是代码量太大尤其是重复单元较多的时候,一个例子如下: weight = tf.get_variable('weig...原创 2018-07-29 22:12:54 · 3037 阅读 · 1 评论 -
获得tensorflow模型的参数量
tensorflow模型训练好之后,通常会保存为.ckpt文件,有时我们想了解一下模型保存了多少参数,这固然可以手动计算,但是速度太慢,这里我写了一个程序可以直接获得模型的参数量,下面是源代码: from tensorflow.python import pywrap_tensorflow import os import numpy as np model_dir = "models...原创 2018-08-11 17:25:58 · 9997 阅读 · 8 评论