o Ne35@sdZddlZddlZddlmmZddlZddlmZddlm Z ddlm Z ddl m Z ddl mZddlmZdd lmZdd lmZdd lmZd d Zd(ddZddZd)ddZd*ddZddZd)ddZddZddZ d d!Z!d"d#Z"d$d%Z#d&d'Z$dS)+z$Utils related to keras model saving.N)backend)losses) optimizers)base_layer_utils) optimizer_v1) serialization) version_utils)ask_to_proceed_with_overwrite) tf_loggingcCs t|ddrdd|jDSdS)aPConvert metrics from a Keras model `compile` API to dictionary. This is used for converting Keras models to Estimators and SavedModels. Args: model: A `tf.keras.Model` object. Returns: Dictionary mapping metric names to metric instances. May return `None` if the model does not contain any metrics. Z_compile_metricsNcSsi|]}|j|qS)name).0mr r PD:\Projects\ConvertPro\env\Lib\site-packages\keras/saving/legacy/saving_utils.py 4sz)extract_model_metrics..)getattrZ_compile_metric_functionsmodelr r rextract_model_metrics$s rFcCs&|j| d}|dur dSt|}|S)atInspect model to get its input signature. The model's input signature is a list with a single (possibly-nested) object. This is due to the Keras-enforced restriction that tensor inputs must be passed in as the first argument. For example, a model with input {'feature1': , 'feature2': } will have input signature: [{'feature1': TensorSpec, 'feature2': TensorSpec}] Args: model: Keras Model object. keep_original_batch_size: A boolean indicating whether we want to keep using the original batch size or set it to None. Default is `False`, which means that the batch dim of the returned input signature will always be set to `None`. Returns: A tuple containing `(args, kwargs)` TensorSpecs of the model call function inputs. `kwargs` does not contain the `training` argument. )Z dynamic_batchN)NN)Z save_spec_enforce_names_consistency)rZkeep_original_batch_sizeZ input_specsr r rmodel_call_inputs8s rcCs.t|tjjrtd|dtd|d)NzModel z cannot be saved because the input shape is not available. Please specify an input shape either by calling `build(input_shape)` directly, or by calling the model on actual data using `Model()`, `Model.fit()`, or `Model.predict()`.a cannot be saved either because the input shape is not available or because the forward pass of the model is not defined.To define a forward pass, please override `Model.call()`. To specify an input shape, either call `build(input_shape)` directly, or call the model on actual data using `Model()`, `Model.fit()`, or `Model.predict()`. If you have a custom training step, please make sure to invoke the forward pass in train step through `Model.__call__`, i.e. `model(inputs)`, as opposed to `model.call()`.) isinstancekerasmodelsZ Sequential ValueErrorrr r rraise_model_input_errorVs  rcsn|durtjtjjjrjj}|r|}i}nt\}}|dur&ttjfdd}|j |i|S)aTrace the model call to create a tf.function for exporting a Keras model. Args: model: A Keras model. input_signature: optional, a list of tf.TensorSpec objects specifying the inputs to the model. Returns: A tf.function wrapping the model's call function with input signatures set. Raises: ValueError: if input signature cannot be inferred from the model. Ncsjjdd||dd\}}tjddddd|i|}Wdn1s+wYj}|durBddlm}||}t j |}d d t ||DS) z._wrapped_model..) Z _call_specZ set_arg_valuerZ call_contextZenter output_names keras.enginerZcreate_pseudo_output_namestfnestflattenzip)argskwargsoutputsrrrr r_wrapped_models      z(trace_model_call.._wrapped_model) rcallr!Z __internal__functionFunctioninput_signaturerrZget_concrete_function)rr,Z model_argsZ model_kwargsr(r rrtrace_model_callms r-Tc Cs ddlm}ddlm}d|jji}z||d<Wnty0}z |r&|WYd}~nd}~wwtt |t |d}|j r|rt |j t jrOtd|S|jr|jd d }|d dt||d <t |j |jrotd tj|j j|j d} | |d d<|S)z3Returns a dictionary containing the model metadata.r) __version__) optimizer_v2 class_nameconfigN) keras_versionr model_configa<TensorFlow optimizers do not make it possible to access optimizer attributes or optimizer state after instantiation. As a result, we cannot save the optimizer as part of the model save file. You will have to compile your model again after loading it. Prefer using a Keras optimizer instead (see keras.io/optimizers).F)Z user_metrics optimizertraining_configzOptimizers loaded from a SavedModel cannot be saved. If you are calling `model.save` or `tf.keras.models.save_model`, please set the `include_optimizer` option to `False`. For `tf.saved_model.save`, delete the optimizer from the model.)r0r1optimizer_config)rr.Zkeras.optimizers.optimizer_v2r/ __class____name__Z get_configNotImplementedErrordictstrrr4rrZ TFOptimizerloggingwarningZ_compile_was_calledZ_get_compile_argspop_serialize_nested_configZRestoredOptimizerutilsZget_registered_name) rZinclude_optimizerZrequire_configr2r/r3emetadatar5r6r r rmodel_metadatasN     "    rCcCs|s tj|r t|SdS)z3Returns whether the filepath should be overwritten.T)ospathisfiler )filepath overwriter r rshould_overwritesrIc Cs|duri}tj|T|d}t|}d}|dd}|dur(ttj|}d}|dd}|dur9tt|}d}|dd} | durJtt| }t |drS|dnd} |d} Wdn1scwYt ||||| | dS) z4Return model.compile arguments from training config.Nr6lossmetricsweighted_metricssample_weight_mode loss_weights)r4rJrKrLrNrM) rr@ZCustomObjectScoper deserializeget_deserialize_nested_configr_deserialize_metrichasattrr:) r5Zcustom_objectsr6r4rJZ loss_configrKZmetrics_configrLZweighted_metrics_configrMrNr r r!compile_args_from_training_configsF       !rTcstdd}|dur dS||r|St|tr"fdd|DSt|ttfr2fdd|DStd|d ) z=Deserializes arbitrary Keras `config` using `deserialize_fn`.cSs(t|tr d|vr dSt|trdSdS)Nr0TF)rr:r;objr r r_is_single_objects  z5_deserialize_nested_config.._is_single_objectNcsi|] \}}|t|qSr rQ)r kvdeserialize_fnr rr s z._deserialize_nested_config..csg|]}t|qSr rX)r rVr[r r %s z._deserialize_nested_config..zrSaved configuration not understood. Configuration should be a dictionary, string, tuple or list. Received: config=.)rr:itemstuplelistr)r\r1rWr r[rrQs$   rQcCsdd}tj||S)z/Serialized a nested structure of Keras objects.cSst|r t|S|SN)callablerZserialize_keras_objectrUr r r _serialize_fn2s z/_serialize_nested_config.._serialize_fn)r!r" map_structure)r1rdr r rr?/sr?cCs"ddlm}|dvr |S||S)z7Deserialize metrics, leaving special strings untouched.r)rK)ZaccuracyaccZ crossentropyZce)rrKrO)Z metric_configZmetrics_moduler r rrR:s  rRcs`dddd}tj|}tfdd|Do$tfdd|D }|r.tj||}|S)z5Enforces that either all specs have names or none do.cSs|dup t|do |jduSNr )rSr specr r r _has_nameIsz-_enforce_names_consistency.._has_namecSst|}t|dr d|_|Srg)copydeepcopyrS_namerhr r r _clear_nameLs  z/_enforce_names_consistency.._clear_namec3s|]}|VqdSrbr )r srjr r Ssz-_enforce_names_consistency..)r!r"r#anyallre)specsrnZ flat_specsZname_inconsistencyr rprrFs " rcCspt|s4|jdur6z|jjs|j|j|jjs&|j|j|jWdSWdStdYdSdSdS)NzCompiled the loaded model, but the compiled metrics have yet to be built. `model.compile_metrics` will be empty until you train or evaluate the model.) rZis_v1_layer_or_modelr'Z compiled_lossZbuiltbuildZcompiled_metricsr<r=rr r rtry_build_compiled_arguments\s  rvcCs|dp|dp|dS)Nz.h5z.kerasz.hdf5)endswith)rGr r ris_hdf5_filepathns rx)Frb)TT)%__doc__rkrDZtensorflow.compat.v2compatv2r!rrrrr rZkeras.optimizersrZkeras.saving.legacyrZ keras.utilsrZkeras.utils.io_utilsr Ztensorflow.python.platformr r<rrrr-rCrIrTrQr?rRrrvrxr r r rs6            48 0