diff --git a/torchvision/transforms/autoaugment.py b/torchvision/transforms/autoaugment.py index efb0cc90e8d..2c42bc375f6 100644 --- a/torchvision/transforms/autoaugment.py +++ b/torchvision/transforms/autoaugment.py @@ -22,6 +22,7 @@ def _apply_op( shear=[math.degrees(magnitude), 0.0], interpolation=interpolation, fill=fill, + center=[0, 0], ) elif op_name == "ShearY": img = F.affine( @@ -32,6 +33,7 @@ def _apply_op( shear=[0.0, math.degrees(magnitude)], interpolation=interpolation, fill=fill, + center=[0, 0], ) elif op_name == "TranslateX": img = F.affine(