有两种方法
一方法,这里用的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模块里面。