mirror of
https://github.com/Azure/MachineLearningNotebooks.git
synced 2025-12-19 17:17:04 -05:00
37 lines
1010 B
Python
37 lines
1010 B
Python
import os
|
|
import sys
|
|
|
|
|
|
def convert(imgf, labelf, outf, n):
|
|
f = open(imgf, "rb")
|
|
temp = open(labelf, "rb")
|
|
o = open(outf, "w")
|
|
|
|
f.read(16)
|
|
temp.read(8)
|
|
images = []
|
|
|
|
for i in range(n):
|
|
image = [ord(temp.read(1))]
|
|
for j in range(28 * 28):
|
|
image.append(ord(f.read(1)))
|
|
images.append(image)
|
|
|
|
for image in images:
|
|
o.write(",".join(str(pix) for pix in image) + "\n")
|
|
f.close()
|
|
o.close()
|
|
temp.close()
|
|
|
|
|
|
mounted_input_path = sys.argv[1]
|
|
mounted_output_path = sys.argv[2]
|
|
os.makedirs(mounted_output_path, exist_ok=True)
|
|
|
|
convert(os.path.join(mounted_input_path, 'train-images-idx3-ubyte'),
|
|
os.path.join(mounted_input_path, 'train-labels-idx1-ubyte'),
|
|
os.path.join(mounted_output_path, 'mnist_train.csv'), 60000)
|
|
convert(os.path.join(mounted_input_path, 't10k-images-idx3-ubyte'),
|
|
os.path.join(mounted_input_path, 't10k-labels-idx1-ubyte'),
|
|
os.path.join(mounted_output_path, 'mnist_test.csv'), 10000)
|