資源簡介
pytorch編寫的cifar10數(shù)據(jù)集分類程序,利用LeNet進(jìn)行分類
代碼片段和文件信息
#?-*-?coding:utf-8?-*-
import?torch
import?torchvision
import?torchvision.transforms?as?transforms
import?numpy?as?np
transform=transforms.Compose([transforms.ToTensor()transforms.Normalize((0.50.50.5)(0.50.50.5))])
trainset=torchvision.datasets.CIFAR10(root=‘./data‘train=Truedownload=Falsetransform=transform)
trainloader=torch.utils.data.DataLoader(trainsetbatch_size=5shuffle=Truenum_workers=2)
testset=torchvision.datasets.CIFAR10(root=‘./data‘train=Falsedownload=Falsetransform=transform)
testloader=torch.utils.data.DataLoader(testsetbatch_size=5shuffle=Falsenum_workers=2)
classes=(‘plane‘‘a(chǎn)utomobile‘‘bird‘‘cat‘‘deer‘‘dog‘‘frog‘‘horse‘‘ship‘‘truck‘)
#?import?matplotlib.pyplot?as?plt
#?import?numpy?as?np
#?def?imshow(img):
#?????img=img/2+0.5
#?????nimg=img.num
評論
共有 條評論