googlenet.py 文件源码

def build_inception_module(self, name, input_layer, nfilters):
        # nfilters: (pool_proj, 1x1, 3x3_reduce, 3x3, 5x5_reduce, 5x5)
        net = dict()
        net['pool'] = PoolLayerDNN(input_layer, pool_size=3, stride=1, pad=1)
        net['pool_proj'] = ConvLayer(
            net['pool'], nfilters[0], 1, flip_filters=False)

        net['1x1'] = ConvLayer(input_layer, nfilters[1], 1, flip_filters=False)

        net['3x3_reduce'] = ConvLayer(
            input_layer, nfilters[2], 1, flip_filters=False)
        net['3x3'] = ConvLayer(
            net['3x3_reduce'], nfilters[3], 3, pad=1, flip_filters=False)

        net['5x5_reduce'] = ConvLayer(
            input_layer, nfilters[4], 1, flip_filters=False)
        net['5x5'] = ConvLayer(
            net['5x5_reduce'], nfilters[5], 5, pad=2, flip_filters=False)

        net['output'] = ConcatLayer([
            net['1x1'],
            net['3x3'],
            net['5x5'],
            net['pool_proj'],
            ])

        return {'{}/{}'.format(name, k): v for k, v in net.items()}