13hannes11/situr
1
0
Fork 0
mirror of https://github.com/13hannes11/situr.git synced 2024-09-03 20:50:58 +02:00
Code Issues Projects Releases Wiki Activity

add documentation for transformation

Browse Source
This commit is contained in:
Hannes F. Kuchelmeister
2021-07-22 11:38:38 +02:00
parent b32451c73c
commit 515abcc187
1 changed files with 13 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
situr/transformation/transformation.py
View File

@@ -2,7 +2,10 @@ import abc
import numpy as np import numpy as np
import scipy import scipy
class Transform: class Transform:
"""Abstract class representing a transformatio that can be performed on an image.
"""
__metaclass__ = abc.ABCMeta __metaclass__ = abc.ABCMeta
@abc.abstractmethod @abc.abstractmethod
@@ -12,13 +15,21 @@ class Transform:
class IdentityTransform(Transform): class IdentityTransform(Transform):
"""A transformation that does not change the image.
It inherits from Transform.
"""
def apply_tranformation(self, img: np.ndarray) -> np.ndarray: def apply_tranformation(self, img: np.ndarray) -> np.ndarray:
return img return img
class ScaleRotateTranslateTransform(Transform): class ScaleRotateTranslateTransform(Transform):
def __init__(self, transform_matrix: np.ndarray, scale: int = 1, offset: np.array = np.array([0, 0])): def __init__(self,
"""Constructor for a Transformation that supports rotation, translation and scaling on an image transform_matrix: np.ndarray,
scale: int = 1,
offset: np.array = np.array([0, 0])):
"""Constructor for a Transformation that supports rotation, translation and scaling on an
image
Args: Args:
transform_matrix (np.ndarray): A matrix of shape (2,2) transform_matrix (np.ndarray): A matrix of shape (2,2)