network_in_network.py 文件源码

def golbal_average_pooling(self, x):
        """
        golbal average pooling
        :param x: [batch, height, width, channels]
        """
        shapes = x.get_shape().as_list()
        kernel_height = shapes[1]
        kernel_width = shapes[2]
        return slim.avg_pool2d(x, kernel_size=[kernel_height, kernel_width], stride=1, padding='VALID',
                               scope='golbal_average_pooling')