o e @s@dZddlZGdddeZGdddeZddZd d ZdS) zBase classes and functions used by all/most augmenters. This module is planned to contain :class:`imgaug.augmenters.meta.Augmenter` in the future. Added in 0.4.0. Nc@eZdZdZdS) SuspiciousMultiImageShapeWarningz9Warning multi-image inputs that look like a single image.N__name__ __module__ __qualname____doc__r r FD:\Projects\ConvertPro\env\Lib\site-packages\imgaug/augmenters/base.pyr rc@r)!SuspiciousSingleImageShapeWarningz?Warning for single-image inputs that look like multiple images.Nrr r r r r r r cCsT|durdSt|r$|jdkr&|jddvr(tjd|jftddSdSdSdS)N)r aYou provided a numpy array of shape %s as a multi-image augmentation input, which was interpreted as (N, H, W). The last dimension however has value 1 or 3, which indicates that you provided a single image with shape (H, W, C) instead. If that is the case, you should use e.g. augmenter(image=) or augment_image() -- note the singular 'image' instead of 'imageS'. Otherwise your single input image will be interpreted as multiple images of shape (H, W) during augmentation.category)iaZ is_np_arrayndimshapewarnr)Zimagesr r r &_warn_on_suspicious_multi_image_shapess   rcCs\|durdS|jdkr(|jddkr*|jdddkr,tjd|jftddSdSdSdS) Nr r r)rraYou provided a numpy array of shape %s as a single-image augmentation input, which was interpreted as (H, W, C). The last dimension however has a size of >=32, which indicates that you provided a multi-image array with shape (N, H, W) instead. If that is the case, you should use e.g. augmenter(imageS=) or augment_imageS(). Otherwise your multi-image input will be interpreted as a single image during augmentation.r)rrrrr )imager r r &_warn_on_suspicious_single_image_shape)s* r)rZimgaugr UserWarningrr rrr r r r s