如何从现有冻结的pb模型文件创建新的tensorflow hub模块

import tensorflow as tf
import tensorflow_hub as hub

model_path = 'path/to/frozen_model.pb'
with tf.io.gfile.GFile(model_path, 'rb') as f:
    graph_def = tf.compat.v1.GraphDef()
    graph_def.ParseFromString(f.read())

tf.compat.v1.reset_default_graph()
tf.import_graph_def(graph_def, name='')

input_node = 'input_node_name'
output_node = 'output_node_name'

with tf.compat.v1.Session() as sess:
    input_tensor = sess.graph.get_tensor_by_name(input_node + ':0')
    output_tensor = sess.graph.get_tensor_by_name(output_node + ':0')

hub_module_path = 'path/to/save/hub_module'
spec = hub.create_module_spec(lambda: tf.compat.v1.saved_model.loader.load(sess, ['serve'], model_path))
module = hub.Module(spec, trainable=True)
module.export(hub_module_path, sess)