Source code for pierogis.ingredients.flip
import numpy as np
from .ingredient import Ingredient
[docs]class Flip(Ingredient):
"""
flip pixels about an axis
"""
[docs] def prep(self, axis: int = 0, **kwargs):
"""
:param axis: 0 to flip vertically, 1 to flip horizontally
"""
self.axis = axis
[docs] def cook(self, pixels: np.ndarray):
"""
flip the pixels
"""
flipped_pixels = np.flip(pixels, axis=self.axis)
return flipped_pixels