train_dataset and val_dataset processing is different? why?
Opened this issue · 0 comments
Melonia-H commented
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)