将当前目录下cifar-10-batches-py转换成train图片和test图片，并同时生成train.txt 和 test.txt

唯相遇

于 2017-04-11 23:13:58 发布

阅读量1.1k

点赞数

CC 4.0 BY-SA版权

分类专栏： caffe

本文链接：https://blog.youkuaiyun.com/ls20121006/article/details/70136156

这篇博客介绍了如何将CIFAR-10数据集的cifar-10-batches-py文件拆分为训练集和测试集，同时创建train.txt和test.txt文件以供后续训练和验证使用。详细步骤包括数据读取、拆分及文件生成过程。

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

from __future__ import print_function
import matplotlib.pyplot as pyplot
import PIL.Image as Image
import pickle
import numpy as np
import random
import os,re

def unpickle(file):
    # data：a 10000x3072 numpy array of uint8s. Each row of the array stores a 32x32 colour image.
    # The first 1024 entries contain the red channel values, the next 1024 the green,