a SicA@sdZddlZddlZddlZddlmZddlmZddlm Z ddl m Z da dZ dZd Zd Zd Zd Zd ZdZe dd#ddZejedede_e dd$ddZddZddZd%ddZdd Zd!d"ZdS)&z@Utilities for ImageNet data preprocessing & prediction decoding.N) activations)backend) data_utils) keras_exportzUhttps://storage.googleapis.com/download.tensorflow.org/data/imagenet_class_index.jsonat Preprocesses a tensor or Numpy array encoding a batch of images. Usage example with `applications.MobileNet`: ```python i = tf.keras.layers.Input([None, None, 3], dtype = tf.uint8) x = tf.cast(i, tf.float32) x = tf.keras.applications.mobilenet.preprocess_input(x) core = tf.keras.applications.MobileNet() x = core(x) model = tf.keras.Model(inputs=[i], outputs=[x]) image = tf.image.decode_png(tf.io.read_file('file.png')) result = model(image) ``` Args: x: A floating point `numpy.array` or a `tf.Tensor`, 3D or 4D with 3 color channels, with values in the range [0, 255]. The preprocessed data are written over the input data if the data types are compatible. To avoid this behaviour, `numpy.copy(x)` can be used. data_format: Optional data format of the image tensor/array. Defaults to None, in which case the global setting `tf.keras.backend.image_data_format()` is used (unless you changed it, it defaults to "channels_last").{mode} Returns: Preprocessed `numpy.array` or a `tf.Tensor` with type `float32`. {ret} Raises: {error} a mode: One of "caffe", "tf" or "torch". Defaults to "caffe". - caffe: will convert the images from RGB to BGR, then will zero-center each color channel with respect to the ImageNet dataset, without scaling. - tf: will scale pixels between -1 and 1, sample-wise. - torch: will scale pixels between 0 and 1 and then will normalize each channel with respect to the ImageNet dataset. zE ValueError: In case of unknown `mode` or `data_format` argument.z; ValueError: In case of unknown `data_format` argument.zH The inputs pixel values are scaled between -1 and 1, sample-wise.z The input pixels values are scaled between 0 and 1 and each channel is normalized with respect to the ImageNet dataset.z The images are converted from RGB to BGR, then each color channel is zero-centered with respect to the ImageNet dataset, without scaling.z2keras.applications.imagenet_utils.preprocess_inputcaffecCsj|dvrtd||dur(t}n|dvr>td|t|tjrXt|||dSt|||dSdS)z@Preprocesses a tensor or Numpy array encoding a batch of images.>rtorchtfzDExpected mode to be one of `caffe`, `tf` or `torch`. Received: mode=N>channels_first channels_lastz]Expected data_format to be one of `channels_first` or `channels_last`. Received: data_format=) data_formatmode) ValueErrorrimage_data_format isinstancenpndarray_preprocess_numpy_input_preprocess_symbolic_input)xr r r]/var/www/html/django/DPS/env/lib/python3.9/site-packages/keras/applications/imagenet_utils.pypreprocess_inputgs"  r)r reterrorz4keras.applications.imagenet_utils.decode_predictionscst|jdks|jddkr.tdt|jtdurztjdtddd }t|}t |aWdn1sp0Yg}|D]N | dddd }fd d |D}|j d ddd| |q|S)aDecodes the prediction of an ImageNet model. Args: preds: Numpy array encoding a batch of predictions. top: Integer, how many top-guesses to return. Defaults to 5. Returns: A list of lists of top class prediction tuples `(class_name, class_description, score)`. One list of tuples per sample in batch input. Raises: ValueError: In case of invalid shape of the `pred` array (must be 2D). izx`decode_predictions` expects a batch of predictions (i.e. a 2D array of shape (samples, 1000)). Found array with shape: Nzimagenet_class_index.jsonmodelsZ c2c37ea517e94d9795004a39431a14cb) cache_subdir file_hashcs&g|]}ttt||fqSr)tuple CLASS_INDEXstr).0ipredrr z&decode_predictions..cSs|dS)Nrr)rrrrr*z$decode_predictions..T)keyreverse)lenshaper r$r#rget_fileCLASS_INDEX_PATHopenjsonloadargsortsortappend)predstopfpathfresultsZ top_indicesresultrr'rdecode_predictionss, ( r>cCs.t|jjtjs"|jtdd}|dkr>|d}|d8}|S|dkr`|d}gd}gd }n^|d kr|jd kr|d d d df}q|d d d d d df}n|dd d d f}gd}d }|d kr|jd kr|dd d d d f|d8<|dd d d d f|d8<|dd d d d f|d8<|d ur|dd d d d f|d<|dd d d d f|d<|dd d d d f|d<n|d d dd d d d f|d8<|d d dd d d d f|d8<|d d dd d d d f|d8<|d ur*|d d dd d d d f|d<|d d dd d d d f|d<|d d dd d d d f|d<n|d|d8<|d|d8<|d|d8<|d ur*|d|d<|d|d<|d|d<|S)aPreprocesses a Numpy array encoding a batch of images. Args: x: Input array, 3D or 4D. data_format: Data format of the image array. mode: One of "caffe", "tf" or "torch". - caffe: will convert the images from RGB to BGR, then will zero-center each color channel with respect to the ImageNet dataset, without scaling. - tf: will scale pixels between -1 and 1, sample-wise. - torch: will scale pixels between 0 and 1 and then will normalize each channel with respect to the ImageNet dataset. Returns: Preprocessed Numpy array. F)copyr_@?ro@g ףp= ?gv/?gCl?gZd;O?gy&1?g?r Nr!.gjtY@g`"1]@gQ^@rrr).r).r).r) issubclassdtypetyperfloatingastyperfloatxndim)rr r meanstdrrrrsR    """ ""$((( ((* rcCs:|dkr|d}|d8}|S|dkr>|d}gd}gd}nb|dkrt|d krh|d d d d f}q|d d d d d d f}n|d d d d f}gd }d }tt| }t|t|krtj|t|t||d}nt|||}|d ur6tjt|t|d}|dkr.t|d}||}|S)aPreprocesses a tensor encoding a batch of images. Args: x: Input tensor, 3D or 4D. data_format: Data format of the image tensor. mode: One of "caffe", "tf" or "torch". - caffe: will convert the images from RGB to BGR, then will zero-center each color channel with respect to the ImageNet dataset, without scaling. - tf: will scale pixels between -1 and 1, sample-wise. - torch: will scale pixels between 0 and 1 and then will normalize each channel with respect to the ImageNet dataset. Returns: Preprocessed tensor. rr@rArrBrCrDr rENr!.rF)r )rH)r!rr) rrMconstantrarrayrHbias_addcastreshape)rr r rNrO mean_tensorZ std_tensorrrrrs:    rcCsp|dkr|rt|dkr|dkrZ|ddvrJtjdt|dddd |d||f}q|d dvrtjdt|d ddd |||d f}n|dkrd||f}n ||df}|dkr|r|d ur||krtd |d ||S|r.|dkr|d ur,t|dkrtd|ddkr>|dkr>td|d|dd urZ|d|ksv|dd ur,|d|kr,td|d|d|n|d urNt|dkrtd|d dkr|dkrtd|d|dd ur|d|ks|dd urN|d|krNtd|d|d|n |r:|}n|dkrJd}nd}|rld |vrltd||S)aInternal utility to compute/validate a model's input shape. Args: input_shape: Either None (will return the default network input shape), or a user-provided shape to be validated. default_size: Default input width/height for the model. min_size: Minimum input width/height accepted by the model. data_format: Image data format to use. require_flatten: Whether the model is expected to be linked to a classifier via a Flatten layer. weights: One of `None` (random initialization) or 'imagenet' (pre-training on ImageNet). If weights='imagenet' input channels must be equal to 3. Returns: An integer shape tuple (may include None entries). Raises: ValueError: In case of invalid argument values. imagenetrEr r>rrEz]This model usually expects 1 or 3 input channels. However, it was passed an input_shape with z input channels.r) stacklevelr!NzXWhen setting `include_top=True` and loading `imagenet` weights, `input_shape` should be z. Received: input_shape=z0`input_shape` must be a tuple of three integers.z6The input must have 3 channels; Received `input_shape=`rzInput size must be at least rz; Received: input_shape=)rENN)NNrEz[If `include_top` is True, you should specify a static `input_shape`. Received: input_shape=)r.warningswarnr$r ) input_shape default_sizemin_sizer require_flattenweightsZ default_shaperrrobtain_input_shape6s                    r`cCstdkrdnd}t|||d}t|tr<||f}|ddurNd}n d|ddd|ddf}|dd|ddf}|d|d|df|d|d|dffS)zReturns a tuple for zero-padding for 2D convolution with downsampling. Args: inputs: Input tensor. kernel_size: An integer or tuple/list of 2 integers. Returns: A tuple. r rrrN)rr)rr int_shaperint)inputs kernel_sizeZimg_dim input_sizeadjustZcorrectrrr correct_pads    rgcCs@|dur dSt|}|tdtdhvrrrr`rgrkrrrrs@    $   +IB y