用自己的数据集制作类似于Cifar-10格式的数据集

最新推荐文章于 2023-11-01 17:28:03 发布

原创

最新推荐文章于 2023-11-01 17:28:03 发布 · 3.5k 阅读

33 ·

CC 4.0 BY-SA版权

本文档介绍了如何使用Python3改编的代码来制作类似于Cifar-10格式的数据集，详细步骤包括数据集制作代码、运行结果、生成文件内容的展示，最终确保数据格式与Cifar-10一致。

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

前言：

Cifar-10数据集的制作是基于python2.7来写的，以下是修改后的基于python3的版本的Code。

1.数据集制作代码：

from PIL import Image
import os
import numpy as np
import pickle

data = {}
list1 = []
list2 = []
list3 = []

def img_tra():
    for k in range(0,num):
        currentpath = folder + '/' + imglist[k]
        im = Image.open(currentpath)

        x_s = 32
        y_s = 32

        out = im.resize((x_s,y_s),Image.ANTIALIAS)
        out.save(folder_ad + '/' + str(imglist[k]))

def seplabel(fname):
    filestr = fname.split('.')[0]
    label = int(filestr.split('_')[0])
    return label

def mkcf():