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