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

最新推荐文章于 2024-06-27 10:55:55 发布

原创

最新推荐文章于 2024-06-27 10:55:55 发布 · 1.1k 阅读

2 ·

CC 4.0 BY-SA版权

这篇博客介绍了如何将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,