tf.layers.batch_normalization()的原理

tf.layers.batch_normalization()的原理

毕设需要做一个fpga实现yolov2，需要了解tf.layers.batch_normalization()的底层实现
这篇博文写的非常详细

https://blog.csdn.net/TheHonestBob/article/details/103951083

记录一下我遇到的问题
我使用的代码来自

https://zhuanlan.zhihu.com/p/36902889

载入模型后当我使用：

params = tf.trainable_variables()
print("Trainable variables:------------------------")
for idx, v in enumerate(params):
    print("  param {:3}: {:15}   {}".format(idx, str(v.get_shape()), v.name))

提取参数发现只能看到beta和gamma的值

reader = tf.train.NewCheckpointReader("./yolo2_model/yolo2_coco.ckpt")
dic = reader.get_variable_to_shape_map