Closed lxwkobe24 closed 2 years ago
transform = create_transform( input_size=args.input_size, is_training=True,
mean=mean, std=std, )
You may write your own transform, for example, with only mean and std. For example, following this: https://github.com/pytorch/examples/blob/main/imagenet/main.py
transform = create_transform( input_size=args.input_size, is_training=True,
color_jitter=args.color_jitter,
auto_augment=args.aa,
interpolation=args.train_interpolation,
re_prob=args.reprob,
re_mode=args.remode,
re_count=args.recount,