Hallo zusammen, ich habe mit Tensorflow Keras ein Image detection model erstellt. Das ganze in Google Colab. Wenn ich das model speichere und den gleichen Code wie in Colab zum klassifizieren eines Bildes nutze, erhalte ich einen Fehler (Im Anhang). Der Code:
import tensorflow as tf
import numpy as np
import matplotlib.pyplot as plt
img = [
0,0,0,1,1,1,1,1,1,1,1,1,1,1.... Und so weiter
]
img = np.array(img)
img = img.reshape(1,50,50)
model = tf.keras.models.load_model('ElektroAI_V_1.h5')
probability_model = tf.keras.Sequential([model, tf.keras.layers.Softmax()])
class_names = ["Vertikales Kabel", "Horizontales Kabel", "Verbundene Kreuzung", "Nicht verbundene Kreuzung", "Motor, horizontal angebunden", "Motor, vertikal angebunden", "Stromquelle, horizontal angebunden", "Stromquelle, vertikal angebunden","Widerstand, vertikal angebunden", "Widerstand, horizontal angebunden","led, horizontal angebunden ", "led, vertikal angebunden"]
def calcType(a):
predictions_single = probability_model.predict(a)
return np.argmax(predictions_single[0])
print(class_names[int(calcType(img))])
Der Fehler sollte denke ich darauf hinweisen dass das Model in der Version 2.15.0 erstellt wurde, ich aber die Version 2.16.1 nutze. (Die alte kann ich auch irgendwie nicht installieren.) Was nun?
Vielen Dank im Vorraus,
Mats