第五次上机作业
18029100040
吴程锴
一、作业 34:使用卷积神经网络判别狗的类别:
1.1
代码
1. import torch
2. import torch.nn as nn
3. import torch.utils.data as Data
4. from torch.autograd import Variable
5. import torchvision
6. from torchvision import transforms
7. from torchvision.datasets import ImageFolder
8. from PIL import Image
9. import matplotlib.pyplot as plt
10. import numpy as np
11.
12. LR=0.001
13. EPOCH=10
14. train=0
15.
16. Dog_names=['哈士奇','柯基犬','秋田犬','边境牧羊犬']
17. device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
18. size=256
19. data_transform = transforms.Compose([
20. transforms.Resize(size),
21. transforms.CenterCrop((size, size)),
22. transforms.ToTensor(),
23. transforms.Normalize(
24. mean=[0.5,0.5,0.5],
25. std=[0.5, 0.5, 0.5])
26. ])
27. train_dataset = ImageFolder("DogData/",transform = data_transform)
28. train_loader = Data.DataLoader(dataset=train_dataset, batch_size=10, shuffle=True, n
um_workers=2)
29. img, label = train_dataset.__getitem__(600)
30.
评论0