B лd.@spdZddlZddlmZddlmZddlmZddlmZddl m Z ddl m Z d d Z d d Zd dZdS)zSavedModel utils.N)message) text_format)saved_model_pb2)file_io) constants)compatc Csntjt|ttj}tjt|ttj}t |sXt |sXt d|t }t |rt |d}|}WdQRXy|||Stjk r}zt d|t|fWdd}~XYnXnt |rTt |d}|}WdQRXyt|d||Stjk rP}zt d|t|fWdd}~XYnXnt d|tjtjfdS)aReads the saved_model.pb or saved_model.pbtxt file containing `SavedModel`. Args: saved_model_dir: Directory containing the SavedModel file. Returns: A `SavedModel` protocol buffer. Raises: IOError: If the file does not exist, or cannot be successfully parsed. z%SavedModel file does not exist at: %srbNzCannot parse proto file %s: %s.zutf-8zCannot parse pbtxt file %s: %s.z-SavedModel file does not exist at: %s/{%s|%s})ospathjoinras_bytesrZSAVED_MODEL_FILENAME_PBTXTZSAVED_MODEL_FILENAME_PBrZ file_existsIOErrorrZ SavedModelFileIOreadZParseFromStringr DecodeErrorstrrZMergedecodeZ ParseError)saved_model_dirZ path_to_pbtxtZ path_to_pb saved_modelfZ file_contenter[/var/www/html/venv/lib/python3.7/site-packages/tensorflow/python/tools/saved_model_utils.pyread_saved_models:    ( (rcCs2t|}g}x |jD]}|t|jjqW|S)zRetrieves all the tag-sets available in the SavedModel. Args: saved_model_dir: Directory containing the SavedModel. Returns: List of all tag-sets in the SavedModel, where a tag-set is represented as a list of strings. )r meta_graphsappendlist meta_info_deftags)rrZall_tagsmeta_graph_defrrrget_saved_model_tag_setsMs  r cCstt|}tdd|dD}g}x6|jD],}t|jj}||krH|S|d|q,Wtd|d|dS)aeGets MetaGraphDef from SavedModel. Returns the MetaGraphDef for the given tag-set and SavedModel directory. Args: saved_model_dir: Directory containing the SavedModel to inspect. tag_set: Group of tag(s) of the MetaGraphDef to load, in string format, separated by ','. The empty string tag is ignored so that passing '' means the empty tag set. For tag-set contains multiple tags, all tags must be passed in. Raises: RuntimeError: An error when the given tag-set does not exist in the SavedModel. Returns: A MetaGraphDef corresponding to the tag-set. cSsg|] }|r|qSrr).0tagrrr ssz&get_meta_graph_def..,z%MetaGraphDef associated with tag-set zQ could not be found in the SavedModel. Please use one of the following tag-sets: N) rsetsplitrrrrr RuntimeError)rZtag_setrZ set_of_tags valid_tagsrZmeta_graph_tagsrrrget_meta_graph_def^s  r))__doc__r Zgoogle.protobufrrZtensorflow.core.protobufrZtensorflow.python.lib.iorZtensorflow.python.saved_modelrZtensorflow.python.utilrrr r)rrrrs      2