n12_pepe_zoo.py 文件源码

def build_mod10(opt=adam()):
    n = int(1800)
    in1 = Input((128,), name='x1')

    x1 = fc_block1(in1, n)
    x1 = fc_inception(x1, n)
    x1 = fc_inception(x1, n)

    in2 = Input((1024,), name='x2')

    x2 = fc_block1(in2, n)
    x2 = fc_inception(x2, n)
    x2 = fc_inception(x2, n)

    x = merge([x1, x2], mode='concat', concat_axis=1)

    x = fc_inception(x, n)
    x = fc_inception(x, n)
    x = fc_block1(x, 2000)

    out = Dense(4716, activation='sigmoid', name='output')(x)
    model = Model(input=[in1, in2], output=out)
    model.compile(optimizer=opt, loss='categorical_crossentropy')

    # model.summary()
    plot(model=model, show_shapes=True)
    return model