mnist数据集
数据集下载,推荐不使用
python download.py mnist
直接去
http://yann.lecun.com/exdb/mnist/
下载数据集。同时需要注意的是,解压后
t10k-images.idx3-ubyte中的.改成-。这是main中的设置问题。
main. py
1
python main.py --dataset mnist --input_height=28 --output_height=28 --train
这里在我的py3.5环境中,需要更改main.py中的:
flags.DEFINE_integer(“train_size”, np.inf, “The size of train images [np.inf]”)
为flags.DEFINE_float(“train_size”, np.inf, “The size of train images [np.inf]”)
2
AttributeError: module 'scipy.misc' has no attribute 'imsave'
安装
conda install Pillow