[ 仅 API 调用方式不一致 ]torchvision.transforms.RandomAffine

torchvision.transforms.RandomAffine

torchvision.transforms.RandomAffine(degrees, translate=None, scale=None, shear=None, interpolation=torchvision.transforms.InterpolationMode.InterpolationMode.NEAREST, fill=0, center=None)

paddle.vision.transforms.RandomAffine

paddle.vision.transforms.RandomAffine(degrees, translate=None, scale=None, shear=None, interpolation='nearest', fill=0, center=None, keys=None)

两者功能一致,但调用方式不一致,具体如下:

转写示例

# PyTorch 写法
transform = torchvision.transforms.RandomAffine(
    45,
    (0.1, 0.1),
    (0.5, 1.5),
    20,
    torchvision.transforms.InterpolationMode.BILINEAR,
    255,
    (2, 2),
)

# Paddle 写法
transform = paddle.vision.transforms.RandomAffine(
    45,
    (0.1, 0.1),
    (0.5, 1.5),
    20,
    "bilinear",
    255,
    (2, 2)
)