MNIST图片可视化(Win7+python3.5)

最新推荐文章于 2021-09-19 17:23:37 发布

VIP文章 VOY_KANG

最新推荐文章于 2021-09-19 17:23:37 发布

阅读量645

点赞数

本文链接：https://blog.csdn.net/rono_back/article/details/77799144

版权

将%caffe_root%\data\mnist\get_mnist.sh转换成linux格式，并用cygwin运行（事先装好wget）。自动下载并解压，得到t10k-images-idx3-ubyte，t10k-labels-idx1-ubyte，train-images-idx3-ubyte，train-labels-idx1-ubyte
创建%caffe_root%\data\mnist\mnist_images\test及%caffe_root%\data\mnist\mnist_images\train文件夹，以保存转换后的图片
编写如下python脚本，并运行。在对应文件夹下得到图片文件。

import numpy as np
import struct
from PIL import Image

caffe_root = r"D:/caffe"
caffe_mnist_data = caffe_root + r"/data/mnist/"

test_MNIST_data = caffe_mnist_data + r"t10k-images-idx3-ubyte"
train_MNIST_data = caffe_mnist_data + r"train-images-idx3-ubyte"
test_MNIST_label = caffe_mnist_data + r"t10k-labels-idx1-ubyte"
train_MNIST_label = caffe_mnist_data + r"train-labels-idx1-ubyte"

### LABEL ###
index = 0
print("TEST LABEL DATA:" + test_MNIST_label)
binfile