mnist数据集图片提取出来

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

csdn_1HAO

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

阅读量3.5k

点赞数 2

分类专栏： tensorflow

本文链接：https://blog.csdn.net/caomin1hao/article/details/80590925

版权

tensorflow 专栏收录该内容

36 篇文章 3 订阅

订阅专栏

# -*- coding: UTF-8 -*-

# 把mnist数据集转成图片做测试，图片更为通用

import cv2
import os
from keras.datasets import mnist

import numpy as np
str_1 = 'mnisttrain'
str_2 = 'mnisttest'
if os.path.exists(str_1):
    os.mkdir(str_1)

if os.path.exists(str_2):
    os.mkdir(str_2)
#自动下载mnist数据集
(X_train, Y_train), (X_test, Y_test) = mnist.load_data()

for i in range(0, 59999):  # 迭代 0 到 59999 之间的数字
    fileName = "mnisttrain/"+ str(Y_train[i]) + "_" + str(i) + ".jpg"
    cv2.imwrite(fileName, X_train[i])

for i in range(0, 9999):  # 迭代 0 到 9999 之间的数字
    fileName = "mnisttest/"+ str(Y_test[i]) + "_" + str(i) + ".jpg"
    cv2.imwrite(fileName, X_test[i])