common/data_augmentation/__init__.py

# /*---------------------------------------------------------------------------------------------
#  * Copyright (c) 2022 STMicroelectronics.
#  * All rights reserved.
#  * This software is licensed under terms that can be found in the LICENSE file in
#  * the root directory of this software component.
#  * If no LICENSE file comes with this software, it is provided AS-IS.
#  *--------------------------------------------------------------------------------------------*/

from .random_utils import grayscale_not_supported, check_dataaug_argument, remap_pixel_values_range, apply_change_rate
from .random_misc import random_blur, random_gaussian_noise, check_random_crop_arguments, random_crop, random_jpeg_quality
from .random_erasing import random_rectangle_erasing
from .random_color import random_contrast, random_brightness, random_gamma, random_hue, random_saturation, random_value, \
                                            random_hsv, random_rgb_to_hsv, random_rgb_to_grayscale, random_sharpness, random_posterize, \
                                            random_invert, random_solarize, random_autocontrast
from .random_affine_utils import check_fill_and_interpolation, transform_images, get_flip_matrix, get_translation_matrix, \
                                                   get_rotation_matrix, get_shear_matrix, get_zoom_matrix
from .random_affine import random_flip, random_translation, random_rotation, random_shear, random_zoom