Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

add direction support to shear #199

Open
wants to merge 1 commit into
base: master
Choose a base branch
from
Open

add direction support to shear #199

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 3 additions & 2 deletions Augmentor/Operations.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1154,7 +1154,7 @@ class Shear(Operation):

For sample code with image examples see :ref:`shearing`.
"""
def __init__(self, probability, max_shear_left, max_shear_right):
def __init__(self, probability, max_shear_left, max_shear_right, directions=['x','y']):
"""
The shearing is randomised in magnitude, from 0 to the
:attr:`max_shear_left` or 0 to :attr:`max_shear_right` where the
Expand All @@ -1171,6 +1171,7 @@ def __init__(self, probability, max_shear_left, max_shear_right):
:type max_shear_right: Integer
"""
Operation.__init__(self, probability)
self.directions = directions
self.max_shear_left = max_shear_left
self.max_shear_right = max_shear_right

Expand Down Expand Up @@ -1227,7 +1228,7 @@ def perform_operation(self, images):
# any of the affine transformation matrices, seen here:
# https://en.wikipedia.org/wiki/Transformation_matrix#/media/File:2D_affine_transformation_matrix.svg

directions = ["x", "y"]
directions = self.directions # ["x", "y"]
direction = random.choice(directions)

def do(image):
Expand Down
7 changes: 5 additions & 2 deletions Augmentor/Pipeline.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1484,7 +1484,7 @@ def skew(self, probability, magnitude=1):
skew_type="RANDOM",
magnitude=magnitude))

def shear(self, probability, max_shear_left, max_shear_right):
def shear(self, probability, max_shear_left, max_shear_right, directions=['x','y']):
"""
Shear the image by a specified number of degrees.

Expand All @@ -1498,6 +1498,8 @@ def shear(self, probability, max_shear_left, max_shear_right):
Cannot be larger than 25 degrees.
:param max_shear_right: The max number of degrees to shear to the
right. Cannot be larger than 25 degrees.
:param directions: The directions to shear along.
Has to be one of ['x','y'], ['x'], and ['y']. Defaults to ['x','y']
:return: None
"""
if not 0 < probability <= 1:
Expand All @@ -1509,7 +1511,8 @@ def shear(self, probability, max_shear_left, max_shear_right):
else:
self.add_operation(Shear(probability=probability,
max_shear_left=max_shear_left,
max_shear_right=max_shear_right))
max_shear_right=max_shear_right,
directions=directions))

def greyscale(self, probability):
"""
Expand Down