tensorflow 实现全局平均池化

有两种方法

一方法,这里用的tf.nn。可以用tf.layers代替。

 p05 = tf.nn.avg_pool2d(conv10,ksize=[1,conv10.get_shape().as_list()[1],conv10.get_shape().as_list()[1],1],strides=[1,1,1,1],padding=VALID',name='GAP')

二方法

 p05 = tf.reduce_mean(conv10, [1, 2], keep_dims=True, name='GAP')

我更喜欢第一种,在量化模型时候,第二种可能会报错。不去深究原因,只能说tf某些东西做的还不成熟,比如pruning模块里面。

猜你喜欢

转载自blog.csdn.net/weixin_39875161/article/details/108674314