a +=icB-@sdZddlmZddlmZddlmZddlZddlZddlZddl m Z ze dj Z Wne ypeZ Yn0dd lmZmZmZGd d d e ZGd d d ZdS)z9Utilities for real-time data augmentation on image data. )absolute_import)division)print_functionN)get_keras_submoduleutils) array_to_img img_to_arrayload_imgc@sleZdZdZdZddZddZddZd d Zd d Z d dZ ddZ ddZ ddZ ddZddZdS)IteratoraBase class for image data iterators. Every `Iterator` must implement the `_get_batches_of_transformed_samples` method. # Arguments n: Integer, total number of samples in the dataset to loop over. batch_size: Integer, size of a batch. shuffle: Boolean, whether to shuffle the data between epochs. seed: Random seeding for data shuffling. )ZpngZjpgZjpegZbmpppmZtifZtiffcCsB||_||_||_||_d|_d|_t|_d|_ | |_ dSNr) n batch_sizeseedshuffle batch_indextotal_batches_seen threadingLocklock index_array _flow_indexindex_generator)selfrrrrrs/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/keras_preprocessing/image/iterator.py__init__$s zIterator.__init__cCs(t|j|_|jr$tj|j|_dSN)npZarangerrrrandomZ permutationrrrr_set_index_array/szIterator._set_index_arraycCs|t|kr"tdj|t|d|jdur@tj|j|j|jd7_|jdur`||j|j ||j |d}| |S)NzEAsked to retrieve element {idx}, but the Sequence has length {length})idxlengthr) len ValueErrorformatrrr rrr"r#_get_batches_of_transformed_samples)rr#rrrr __getitem__4s     zIterator.__getitem__cCs|j|jd|jS)Nr)rrr!rrr__len__CszIterator.__len__cCs |dSr)r"r!rrr on_epoch_endFszIterator.on_epoch_endcCs d|_dSr )rr!rrrresetIszIterator.resetccs||jdur&tj|j|j|jdkr8||jdkrHd}n|j|j|j}|j||jkrz|jd7_nd|_|jd7_|j |||jVqdS)Nrr) r,rrr rrr"rrr)rZ current_indexrrrrLs   zIterator._flow_indexcCs|Srrr!rrr__iter__bszIterator.__iter__cOs|j|i|Sr)next)rargskwargsrrr__next__gszIterator.__next__cCs:|jt|j}Wdn1s&0Y||S)zGFor python 2.x. # Returns The next batch. N)rr.rr(rrrrrr.js(z Iterator.nextcCstdS)Gets a batch of transformed samples. # Arguments index_array: Array of sample indices to include in batch. # Returns A batch of transformed samples. N)NotImplementedErrorr2rrrr(vs z,Iterator._get_batches_of_transformed_samplesN)__name__ __module__ __qualname____doc__Zwhite_list_formatsrr"r)r*r+r,rr-r1r.r(rrrrr s   r c@sDeZdZdZddZddZeddZedd Zed d Z d S) BatchFromFilesMixinz{Adds methods related to getting batches from filenames It includes the logic to transform image files to batches. c Cs(||_t||_|dvr$td|d||_||_|jdkr`|jdkrR|jd|_qd|j|_nT|jdkr|jdkr|jd|_qd|j|_n$|jdkr|jd |_n d |j|_||_||_||_ | |_ |d ur|jj } |d krd | f} n"|d kr| df} ntd|fnd } | |_ ||_ d S)a Sets attributes to use later for processing files into a batch. # Arguments image_data_generator: Instance of `ImageDataGenerator` to use for random transformations and normalization. target_size: tuple of integers, dimensions to resize input images to. color_mode: One of `"rgb"`, `"rgba"`, `"grayscale"`. Color mode to read images. data_format: String, one of `channels_first`, `channels_last`. save_to_dir: Optional directory where to save the pictures being yielded, in a viewable format. This is useful for visualizing the random transformations being applied, for debugging purposes. save_prefix: String prefix to use for saving sample images (if `save_to_dir` is set). save_format: Format to use for saving sample images (if `save_to_dir` is set). subset: Subset of data (`"training"` or `"validation"`) if validation_split is set in ImageDataGenerator. interpolation: Interpolation method used to resample the image if the target size is different from that of the loaded image. Supported methods are "nearest", "bilinear", and "bicubic". If PIL version 1.1.3 or newer is installed, "lanczos" is also supported. If PIL version 3.4.0 or newer is installed, "box" and "hamming" are also supported. By default, "nearest" is used. >Z grayscalergbargbzInvalid color mode:z); expected "rgb", "rgba", or "grayscale".r:Z channels_last)r;))rNZ validationrZtrainingrz;Invalid subset name: %s;expected "training" or "validation")image_data_generatortuple target_sizer& color_mode data_format image_shape save_to_dir save_prefix save_format interpolationZ_validation_splitsplitsubset) rr>r@rArBrDrErFrIrGZvalidation_splitrHrrrset_processing_attrssJ$           z(BatchFromFilesMixin.set_processing_attrsc stjtf|j|jd}|j}tD]x\}}t|||j|j |j d}t ||j d}t |drl||jr|j|j}|j||}|j|}|||<q*|jrtD]P\}}t|||j dd}dj|j|tjd|jd } |tj|j| q|jd kr| } n|jd vr`tj!t||jd} tD]\}} |j"| | |<qDn|jd krtjt|t|j#f|jd} tD]\}} d | ||j"| f<qn>|jdkrԇfdd|j$D} n|jdkr|j$} n|S|j%dur|| fS|| |j%fSdS)r3)dtype)rAr@rG)rBcloseT)scalez {prefix}_{index}_{hash}.{format}gcA)prefixindexhashr'input>sparsebinaryZ categoricalg?Z multi_outputcsg|] }|qSrr).0outputrrr zKBatchFromFilesMixin._get_batches_of_transformed_samples..rawN)&rZzerosr%rCrK filepaths enumerater rAr@rGr rBhasattrrLr>Zget_random_transformshapeZapply_transformZ standardizerDrr'rEr randintrFsaveospathjoinZ class_modecopyemptyclassesZ class_indiceslabels sample_weight) rrZbatch_xrZijZimgxparamsfnameZbatch_yZ n_observationrrVrr(s\             z7BatchFromFilesMixin._get_batches_of_transformed_samplescCstdt|jdS)z%List of absolute paths to image filesz;`filepaths` property method has not been implemented in {}.Nr4r'typer5r!rrrrZs zBatchFromFilesMixin.filepathscCstdt|jdS)z!Class labels of every observationz8`labels` property method has not been implemented in {}.Nrmr!rrrrfs zBatchFromFilesMixin.labelscCstdt|jdS)Nz?`sample_weight` property method has not been implemented in {}.rmr!rrrrg"s z!BatchFromFilesMixin.sample_weightN) r5r6r7r8rJr(propertyrZrfrgrrrrr9sM=  r9)r8 __future__rrrr`rnumpyrZkeras_preprocessingrSequenceZ IteratorType ImportErrorobjectrrr r r r9rrrrs      l