MNIST數據集會有二進制文件,如果需要將二進制文件解析出圖片並保存,可以參考下面代碼。
其中pic文件夾下的存儲的是目標圖片,train-images.idx3-ubyte是數據集的圖片文件,train-labels.idx1-ubyte是標籤的二進制文件
下面的是transfer.py的內容:
import numpy as np
import struct
from PIL import Image
import os
# 從一個二進制或者其他文件中讀取的內容進行解析操作
data_file = 'train-images.idx3-ubyte' #需要修改的路徑
# It's 47040016B, but we should set to 47040000B
data_file_size = 47040016
data_file_size = str(data_file_size - 16) + 'B'
data_buf = open(data_file, 'rb').read()
# >表示大端模式 格式符"i"表示轉換爲int,'ii'表示有兩個int變量
magic, numImages, numRows, numColumns = struct.unpack_from(
'>IIII', data_buf, 0)
datas = struct.unpack_from(
'>' + data_file_size, data_buf, struct.calcsize('>IIII'))
datas = np.array(datas).astype(np.uint8).reshape(
numImages, 1, numRows, numColumns)
label_file = 'train-labels.idx1-ubyte' #需要修改的路徑
# It's 60008B, but we should set to 60000B
label_file_size = 60008
label_file_size = str(label_file_size - 8) + 'B'
label_buf = open(label_file, 'rb').read()
magic, numLabels = struct.unpack_from('>II', label_buf, 0)
labels = struct.unpack_from(
'>' + label_file_size, label_buf, struct.calcsize('>II'))
labels = np.array(labels).astype(np.int64)
datas_root = 'D:/pythonWorkSpace/gzToBmp/pic' #讀取二進制數據需要保存的路徑
if not os.path.exists(datas_root):
os.mkdir(datas_root)
for i in range(10):
file_name = datas_root + os.sep + str(i)
if not os.path.exists(file_name):
os.mkdir(file_name)
for i in range(numLabels):
img = Image.fromarray(datas[i, 0, 0:28, 0:28])
label = labels[i]
file_name = datas_root + os.sep + str(label) + os.sep + \
'mnist_train_' + str(ii) + '.png'
img.save(file_name)