Unverified Commit 8eb9fb17 authored by vfdev's avatar vfdev Committed by GitHub
Browse files

Added center as top-left for shear X/Y ops for autoaugment (#5285)



Fixes #5204
Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
parent 8a16e12f
......@@ -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(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment