aegan_reid.py 文件源码

def feature(aegan, filename):
    import ipdb
    with ipdb.launch_ipdb_on_exception():
        aegan.load(prefix='./samples/reid_aegan/aegan/50')

        paths = map(lambda x: x.strip(), open('protocol/cuhk01-all.txt').readlines())
        x = transform( np.array([load_image(path, (64, 128)) for path in paths]) )
        code = aegan.autoencoder.encoder.predict(x)

    ipdb.set_trace()