CIFAR-10画像データセットの作成

CIFAR-10データセットは画像形式ではありません。自分で作成する必要があります

最初にこれを使用してくださいhttps://blog.csdn.net/ctwy291314/article/details/83864405

画像を作る

次に、独自のラベルを作成します

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])

これまでのところ、データセットを取得できます

おすすめ

転載: blog.csdn.net/zhou_438/article/details/113260549