将mnist数据集转换为JPG图片--附带mnist数据集

最新推荐文章于 2024-07-19 21:54:56 发布

追光的IT男

最新推荐文章于 2024-07-19 21:54:56 发布

阅读量2.2k

点赞数 5

文章标签： struct python

本文链接：https://blog.csdn.net/happyrensh/article/details/103584972

版权

本文详细讲解如何将著名的MNIST手写数字数据集转换为JPG图片格式，便于图像处理和分析。通过Python实现数据读取、转换及保存过程。

摘要由CSDN通过智能技术生成

# coding=utf-8
'''将二进制格式的MNIST数据集转成.jpg图片格式并保存，图片标签包含在图片名中'''
import numpy as np
import cv2
import os
import struct

def save_mnist_to_jpg(mnist_image_file, mnist_label_file, save_dir):
    if 'train' in os.path.basename(mnist_image_file):
        prefix = 'train'
    else:
        prefix = 'test'

    labelIndex = 0
    imageIndex = 0
    i = 0
    lbdata = open(mnist_label_file, 'rb').read()
    magic, nums = struct.unpack_from(">II", lbdata, labelIndex)
    labelIndex += struct.calcsize('>II')

    imgdata = open(mnist_image_file, "rb").read()
    magic, nums, numRows, numColumns = struct.unpack_from('>IIII', imgdata, imageIndex)
    imageIndex += struct.calcsize('>IIII')

    for i in range(nums):
        label = struct.unpack_from('>B', lbdata, labelInde