mnist_siamese_generator_pad.py 文件源码

def do_split():
    if os.path.isdir('train') and os.path.isdir('test'):
        return

    (X_train, y_train), (X_test, y_test) = mnist.load_data()

    os.mkdir('train')
    os.mkdir('test')

    np.savetxt('labels_train.csv', y_train, header='label')
    np.savetxt('labels_test.csv', y_test, header='label')

    for i in xrange(X_train.shape[0]):
        im = Image.fromarray(np.uint8(X_train[i]))
        im.save('train'+str(i)+'.png')

    for i in xrange(X_test.shape[0]):
        im = Image.fromarray(np.uint8(X_test[i]))
        im.save('test'+str(i)+'.png')   


#if __name__ == "__main__":