facebookresearch / ConvNeXt

Code release for ConvNeXt model
MIT License
5.78k stars 696 forks source link

您好,想问一下这段数据增强,如果我不想要的话,请问要怎么去写,因为好像它会跳转到timm.data,我不想要RandomResizedCropAndInterpolation这个操作,您好,感谢您的回答 #65

Closed lxwkobe24 closed 2 years ago

lxwkobe24 commented 2 years ago

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,

        mean=mean,
        std=std,
    )
liuzhuang13 commented 2 years ago

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