
tensorflow
yjl9122
做出一些事,影响一些人
展开
-
tensorflow-4D特征图放缩,tensorboard初使用
tensorflow对于图像的放缩可以参照https://github.com/tensorflow/tensorflow/blob/4c192f060cf9ff897911d240c140299d6db257b6/tensorflow/python/ops/image_ops_impl.py在其中有个resize_images函数,该函数可以实现3D和4Dtensor的形状变化 image=原创 2017-07-11 21:37:24 · 1388 阅读 · 0 评论 -
tensorflow--使用freeze_graph.py将ckpt转为pb文件
这节是关于tensorflow的Freezing,字面意思是冷冻,可理解为整合合并;整合什么呢,就是将模型文件和权重文件整合合并为一个文件,主要用途是便于发布。 官方解释可参考:https://www.tensorflow.org/extend/tool_developers/#freezing 这里我按我的理解翻译下,不对的地方请指正: 有一点令我们为比较困惑的是,tensorflow在训练原创 2017-10-25 15:05:32 · 42730 阅读 · 25 评论 -
查看tensorflow ckpt文件中的变量名和对应值
from tensorflow.python import pywrap_tensorflow checkpoint_path = os.path.join(model_dir, "model.ckpt") reader = pywrap_tensorflow.NewCheckpointReader(checkpoint_path) var_to_shape_map = reader.g原创 2017-11-28 17:09:15 · 1081 阅读 · 0 评论 -
syncbn在TensorFlow中的实现
在syncbn之前我们先简单介绍一下BN层以及多卡机制 BN层介绍 BN层中有两个可训练参数(beta, gamma),以及两个统计参数(moving mean, moving variance)。训练过程和测试过程,BN层计算方式是不同的。训练过程,beta和gamma与卷积层中的weight是一样参与训练的,然后moving mean与moving variance由当前的batch me...原创 2018-12-02 22:35:24 · 2853 阅读 · 0 评论