Я загружаю предварительно обученную модель классификатора Tensorflow (TF 2.0), используя приведенный ниже код:
with tf.compat.v1.Session(graph=tf.Graph()) as sess:
model=tf.compat.v1.saved_model.loader.load(sess,[tf.compat.v1.saved_model.tag_constants.SERVING],model_path)
print("Tensorflow Classifier Model loaded successfully")
После загрузки, используя ее для вывода:
from tensorflow.keras.preprocessing.image import ImageDataGenerator
test_datagen = ImageDataGenerator(rescale=1./255)
test_generator = test_datagen.flow_from_directory(Test_directory,classes=['A','B'],class_mode="binary",
shuffle=False,target_size=(224,224))
test_generator.reset()
probs = model.predict(test_generator)
Однако я ' m появляется следующая ошибка:
AttributeError: predict
Может ли кто-нибудь помочь мне решить эту проблему. Спасибо.