Use `tf.compat.v1.graph_util.extract_sub_graph`
Traceback (most recent call last):
File "zsmeif.py", line 107, in <module>
pb_weights_animation = WeightsAnimation(tflitepath, model_path)
File "C:\Users\Administrator\Desktop\project\FACEGOOD-Audio2Face\code\test\AiSpeech\lib\tensorflow\input_lpc_output_weight.py", line
26, in __init__
self.convert_to_tflite(pb_model_path)
File "C:\Users\Administrator\Desktop\project\FACEGOOD-Audio2Face\code\test\AiSpeech\lib\tensorflow\input_lpc_output_weight.py", line
37, in convert_to_tflite
converter = tf.lite.TFLiteConverter.from_saved_model(pb_model_path)
File "D:\anaconda3\envs\audio2face_env\lib\site-packages\tensorflow_core\lite\python\lite.py", line 762, in from_saved_model
output_arrays, tag_set, signature_key)
File "D:\anaconda3\envs\audio2face_env\lib\site-packages\tensorflow_core\lite\python\convert_saved_model.py", line 206, in freeze_sav
ed_model
frozen_graph_def = util.freeze_graph(sess, in_tensors, out_tensors)
File "D:\anaconda3\envs\audio2face_env\lib\site-packages\tensorflow_core\lite\python\util.py", line 249, in freeze_graph
output_arrays)
File "D:\anaconda3\envs\audio2face_env\lib\site-packages\tensorflow_core\python\util\deprecation.py", line 324, in new_func
return func(*args, **kwargs)
File "D:\anaconda3\envs\audio2face_env\lib\site-packages\tensorflow_core\python\framework\graph_util_impl.py", line 277, in convert_v
ariables_to_constants
inference_graph = extract_sub_graph(input_graph_def, output_node_names)
File "D:\anaconda3\envs\audio2face_env\lib\site-packages\tensorflow_core\python\util\deprecation.py", line 324, in new_func
return func(*args, **kwargs)
File "D:\anaconda3\envs\audio2face_env\lib\site-packages\tensorflow_core\python\framework\graph_util_impl.py", line 197, in extract_s
ub_graph
_assert_nodes_are_present(name_to_node, dest_nodes)
File "D:\anaconda3\envs\audio2face_env\lib\site-packages\tensorflow_core\python\framework\graph_util_impl.py", line 152, in _assert_n
odes_are_present
assert d in name_to_node, "%s is not in graph" % d
AssertionError: StatefulPartitionedCall:1 is not in graph
when I run python zsmeif.py， I get the error：