a guest Mar 23rd, 2019 60 Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
- traindata = datasets.ImageFolder(data_dir + '/train', transforms.ToTensor())
- image_means = torch.stack([t.mean(1).mean(1) for t, c in traindata])
RAW Paste Data