解压和处理mnist数据集和fashion-mnist数据集,将其转换为正常图片格式
# -*- coding:utf-8 -*-
#@Time : 2020/12/30 11:07
#@Author: LCHJ
#@File : savemnist.py.py
'''将二进制格式的MNIST数据集转成.jpg图片格式并保存,图片标签包含在图片名中'''
import numpy as np
import cv2
import os
def save_mnist_to_jpg(mnist_image_file, mnist_label_file, save_dir):
if 'train' in os.path.basename(mnist_image_file):
num_file = 60000
prefix = 'train'
else:
num_file = 10000
prefix = 'test'
with open(mnist_image_file, 'rb') as f1:
image_file = f1.read()
with open(mnist_label_file, 'rb') as f2:
label_file = f2.read()
image_file = image_file[16:]
label_file = label_file[8:]
for i in range(num_file):
label = int(label_file[i])
image_list = [int(item) for item in image_file[i*784:i*784+784]]
imag