Open Melonia-H opened 4 years ago
train_dataset = CASIA( transforms.Compose([ transforms.RandomResizedCrop(img_size), transforms.RandomHorizontalFlip(), transforms.ToTensor(), ColorAugmentation(), normalize, ]),phase_train=True) val_dataset = CASIA( transforms.Compose([ transforms.Resize(int(256 * ratio)), transforms.CenterCrop(img_size), transforms.ToTensor(), normalize, ]),phase_train=False,phase_test=args.phase_test)