augmentations.py 文件源码

def get_advanced_transform(dim, mean_values):
    # loader must be cv2 loader
    swap = (2, 1, 0)
    return transforms.Compose([
        Scale(dim),
        Padding(5),
        RandomCrop(5),
        transforms.ToTensor(),
        transforms.Lambda(lambda x: x.mul(255)),
        RandomHue(),
        SwapChannels(swap),
        transforms.Normalize(mean_values, (1, 1, 1))
    ])