MXNet在Win10下安装好后,可以找到
E:\mxnet-master\example\image-classification下的
train_mnist.py跑一个实例。
- 事先在该目录下建立一个子目录mnist
train-images-idx3-ubyte.gz: training set images (9912422 bytes)
train-labels-idx1-ubyte.gz: training set labels (28881 bytes)
t10k-images-idx3-ubyte.gz: test set images (1648877 bytes)
t10k-labels-idx1-ubyte.gz: test set labels (4542 bytes)
下载这四个文件,存放到mnist目录下
- 由于train_mnist.py是面向linux等系统写的,在win10下要稍作改动
1.数据路径设置
这里写了绝对路径,相对路径还不知道怎么设置。
原来是
parser.add_argument('--data-dir', type=str, default='mnist/',
help='the input data directory')
改为:
default='e:\\mxnet-master\\example\\image-classification\\mnist\\'
2.判断数据路径是否存在
- 命令窗口下运行命令
E:\mxnet-master>python example/image-classification/train_mnist.py
输出训练结果: