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

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

唯相遇

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

阅读量1k

点赞数

分类专栏： caffe

本文链接：https://blog.csdn.net/ls20121006/article/details/70136156

版权

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

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

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,
    # and the final 1024 the blue. The image is stored in row-major order,
    # so that the first 32 entries of the array are the red channel values of the first row of the image.

    # labels：a list of 10000 number