CAE.py 文件源码

def load_data(opts):
    dataDir = opts.inDir  #/data/datasets/MNIST/mnist.pkl
    train,test,val = np.load(dataDir,mmap_mode='r')

    return train[0].reshape(-1,1,28,28).astype(floatX), train[1], test[0].reshape(-1,1,28,28).astype(floatX), test[1], val[0].astype(floatX), val[1]