[ 仅 API 调用方式不一致 ]torchvision.transforms.functional.affine
torchvision.transforms.functional.affine
torchvision.transforms.functional.affine(img, angle, translate, scale, shear, interpolation=torchvision.transforms.functional.NEREAST, fill=None, center=None)
paddle.vision.transforms.affine
paddle.vision.transforms.affine(img, angle, translate, scale, shear, interpolation='nearst', fill=0, center=None)
两者功能一致,但调用方式不一致,具体如下:
转写示例
# PyTorch 写法
result = torchvision.transforms.functional.affine(
img, 30, (1, 1), 1.0, 0, torchvision.transforms.InterpolationMode.NEAREST
)
# Paddle 写法
result = paddle.vision.transforms.affine(
img, 30, (1, 1), 1.0, 0, torchvision.transforms.InterpolationMode.NEAREST
)
