Блог

Я пытаюсь использовать модель для распознавания текста, и у меня есть этот блок кода, который приводит к ошибке: «AttributeError: модуль ‘tensorflow._api.v2.saved_model’ не имеет атрибута ‘loader'», который, как я предполагаю, связан с использованием конструкции TF версии 1, которая больше недействительна в V2

 def load_graph(self, modelFile):    
        graph = tf.Graph()
        # graph_def = tf.compat.v1.GraphDef()
        sess = tf.compat.v1.Session(graph=tf.Graph())
Point of failure -> model = tf.saved_model.loader.load(sess, [tf.saved_model.tag_constants.SERVING], './model')
        graph_def = model.graph_def
        
        with open(modelFile, "rb") as f:
            graph_def.ParseFromString(f.read())
        
        with graph.as_default():
            tf.import_graph_def(graph_def)
        
        return graph
 

Я изменил исходную конструкцию с tf.Session () на tf.compat.v1.Session (), что приводит нас к этой ошибке.
Что было бы эквивалентом для TF V2, который я могу использовать в этом случае?

РЕДАКТИРОВАТЬ: Ранее я пробовал этот вариант, который имел ту же ошибку.

 def load_graph(self, modelFile):
         
        graph = tf.Graph()
        graph_def = tf.GraphDef()
         
        with open(modelFile, "rb") as f:
            graph_def.ParseFromString(f.read())
         
        with graph.as_default():
            tf.import_graph_def(graph_def)
         
        return graph