MNIST数据集转换为图片数据集的样例程序

最新推荐文章于 2022-06-18 09:20:50 发布

皮哩皮哩

最新推荐文章于 2022-06-18 09:20:50 发布

阅读量1.1k

点赞数

分类专栏：神经网络文章标签：深度学习 python tensorflow 机器学习

本文链接：https://blog.youkuaiyun.com/qq_42800834/article/details/109176838

版权

这篇博客介绍如何将MNIST数据集转换为图片格式，以适应特定API的需求。内容包括转换背景、所需依赖（numpy，opencv和MNIST数据集）及详细程序主体。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一、编写背景

因为需要对接一个官方的编程API，本人需要自己按其要求搭建一个神经网络，以尝试调用某模块的工作。我参考了Tensorflow的参考书了解了MNIST数据集，然后我准备把MNIST数据集转换为图片格式，以适应API的要求。
同样，这个程序转化出的图片格式的MNIST数据集和标签集也非常适合初学者第一次搭建网络。

二、基础依赖

numpy，opencv，原始MNIST数据集

三、程序主体

# mnist数据集请自行下载，本程序默认数据集在./dataset的文件夹下
from tensorflow.examples.tutorials.mnist import input_data
import cv2
import numpy as np

mnist = input_data.read_data_sets("./dataset", one_hot=True)
# 原数据集的训练集有55000个样本，在此只提取10000个，按需更改
IM