diff --git "a/chapter6-\345\256\236\346\210\230\346\214\207\345\215\227/data/dataset.py" "b/chapter6-\345\256\236\346\210\230\346\214\207\345\215\227/data/dataset.py" index 7e0cfa23..621e8d88 100644 --- "a/chapter6-\345\256\236\346\210\230\346\214\207\345\215\227/data/dataset.py" +++ "b/chapter6-\345\256\236\346\210\230\346\214\207\345\215\227/data/dataset.py" @@ -1,4 +1,5 @@ # coding:utf8 + import os from PIL import Image from torch.utils import data @@ -45,7 +46,7 @@ def __init__(self, root, transforms=None, train=True, test=False): else: self.transforms = T.Compose([ T.Resize(256), - T.RandomReSizedCrop(224), + T.RandomResizedCrop(224), T.RandomHorizontalFlip(), T.ToTensor(), normalize