def build_transform()

in transfer/datasets.py [0:0]

• 14 lines of code
• 2 McCabe index (conditional complexity)

def build_transform(is_train, args):
    transform_train = transforms.Compose([
        transforms.RandomResizedCrop((args.input_size, args.input_size), scale=(0.05, 1.0)),
        transforms.RandomHorizontalFlip(),
        transforms.ToTensor(),
        transforms.Normalize(IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD),
    ])
    transform_test = transforms.Compose([
        transforms.Resize(int((256 / 224) * args.input_size)),
        transforms.CenterCrop(args.input_size),
        transforms.ToTensor(),
        transforms.Normalize(IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD),
    ])
    return transform_train if is_train else transform_test