|
49 | 49 | "data_dir = 'Cat_Dog_data'\n", |
50 | 50 | "\n", |
51 | 51 | "# TODO: Define transforms for the training data and testing data\n", |
52 | | - "train_transforms = transforms.Compose([transforms.RandomRotation(30),\n", |
53 | | - " transforms.RandomResizedCrop(224),\n", |
54 | | - " transforms.RandomHorizontalFlip(),\n", |
55 | | - " transforms.ToTensor(),\n", |
56 | | - " transforms.Normalize([0.485, 0.456, 0.406],\n", |
57 | | - " [0.229, 0.224, 0.225])])\n", |
58 | | - "\n", |
59 | | - "test_transforms = transforms.Compose([transforms.Resize(255),\n", |
60 | | - " transforms.CenterCrop(224),\n", |
61 | | - " transforms.ToTensor(),\n", |
62 | | - " transforms.Normalize([0.485, 0.456, 0.406],\n", |
63 | | - " [0.229, 0.224, 0.225])])\n", |
| 52 | + "train_transforms =\n", |
| 53 | + "\n", |
| 54 | + "test_transforms =\n", |
64 | 55 | "\n", |
65 | 56 | "# Pass transforms in here, then run the next cell to see how the transforms look\n", |
66 | 57 | "train_data = datasets.ImageFolder(data_dir + '/train', transform=train_transforms)\n", |
|
0 commit comments