CIFAR-10 图片数据集制作

╰+攻爆jí腚メ 2023-01-12 01:59 223阅读 0赞

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

首先使用这篇[https://blog.csdn.net/ctwy291314/article/details/83864405][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])

至此就可以得到数据集

[https_blog.csdn.net_ctwy291314_article_details_83864405]: https://blog.csdn.net/ctwy291314/article/details/83864405