CIFAR-10 图片数据集制作

最新推荐文章于 2024-08-01 14:05:32 发布

喝粥也会胖的唐僧

最新推荐文章于 2024-08-01 14:05:32 发布

阅读量928

点赞数

分类专栏：计算机视觉

本文链接：https://blog.csdn.net/zhou_438/article/details/113260549

版权

计算机视觉专栏收录该内容

28 篇文章 1 订阅

订阅专栏

CIFAR-10数据集是不图片格式，需要自己制作

首先使用这篇https://blog.csdn.net/ctwy291314/article/details/83864405

制作image

然后自己制作label

import os
from shutil import copyfile,move

ls=['airplane', 'automobile', 'bird', 'cat', 'deer', 'dog', 'frog', 'horse', 'ship', 'truck']
for i in ls:
    os.makedirs('train'+'\\'+i,exist_ok=True)
    os.makedirs('test'+'\\'+i,exist_ok=True)

ls_train=os.listdir('train')
ls_train.sort()
ls_test=os.listdir('test')
ls_test.sort()

for j in ls_train:
    m=j.split('_')
    if m[0]=='0':copyfile('train\\'+'\\0_'+m[1],'train\\'+ls[0]+'\\'+m[1])
    elif m[0]=='1':copyfile('train\\'+'\\1_'+m[1],'train\\'+ls[1]+'\\'+m[1])
    elif m[0]=='2':copyfile('train\\'+'\\2_'+m[1],'train\\'+ls[2]+'\\'+m[1])
    elif m[0]=='3':copyfile('train\\'+'\\3_'+m[1],'train\\'+ls[3]+'\\'+m[1])
    elif m[0]=='4':copyfile('train\\'+'\\4_'+m[1],'train\\'+ls[4]+'\\'+m[1])
    elif m[0]=='5':copyfile('train\\'+'\\5_'+m[1],'train\\'+ls[5]+'\\'+m[1])
    elif m[0]=='6':copyfile('train'+'\\6_'+m[1],'train\\'+ls[6]+'\\'+m[1])
    elif m[0]=='7':copyfile('train'+'\\7_'+m[1],'train\\'+ls[7]+'\\'+m[1])
    elif m[0]=='8':copyfile('train'+'\\8_'+m[1],'train\\'+ls[8]+'\\'+m[1])
    elif m[0]=='9':copyfile('train'+'\\9_'+m[1],'train\\'+ls[9]+'\\'+m[1])

for j in ls_test:
    m=j.split('_')
    if m[0]=='0':copyfile('test\\'+'\\0_'+m[1],'test\\'+ls[0]+'\\'+m[1])
    elif m[0]=='1':copyfile('test\\'+'\\1_'+m[1],'test\\'+ls[1]+'\\'+m[1])
    elif m[0]=='2':copyfile('test\\'+'\\2_'+m[1],'test\\'+ls[2]+'\\'+m[1])
    elif m[0]=='3':copyfile('test\\'+'\\3_'+m[1],'test\\'+ls[3]+'\\'+m[1])
    elif m[0]=='4':copyfile('test\\'+'\\4_'+m[1],'test\\'+ls[4]+'\\'+m[1])
    elif m[0]=='5':copyfile('test\\'+'\\5_'+m[1],'test\\'+ls[5]+'\\'+m[1])
    elif m[0]=='6':copyfile('test'+'\\6_'+m[1],'test\\'+ls[6]+'\\'+m[1])
    elif m[0]=='7':copyfile('test'+'\\7_'+m[1],'test\\'+ls[7]+'\\'+m[1])
    elif m[0]=='8':copyfile('test'+'\\8_'+m[1],'test\\'+ls[8]+'\\'+m[1])
    elif m[0]=='9':copyfile('test'+'\\9_'+m[1],'test\\'+ls[9]+'\\'+m[1])

至此就可以得到数据集