0

我在编写我的第一个 NN 程序时遇到了一些麻烦。我基本上想在画布上绘制一个数字并将其提供给应该显示“已识别”数字的 NN。

#create the NN, train with mnist and load img saved from canvas
def Analizza():

    mnist = tf.keras.datasets.mnist
    (x_train, y_train), (x_test, y_test) = mnist.load_data()

    x_train = tf.keras.utils.normalize(x_train, axis=1)

    model = tf.keras.models.Sequential()
    model.add(tf.keras.layers.Flatten())
    model.add(tf.keras.layers.Dense(128, activation=tf.nn.relu, input_shape= x_train.shape[1:]))
    model.add(tf.keras.layers.Dense(128, activation=tf.nn.relu))
    model.add(tf.keras.layers.Dense(10, activation=tf.nn.softmax))
    model.compile(optimizer='adam', loss='sparse_categorical_crossentropy', metrics=['accuracy'])

    model.fit(x_train, y_train, epochs=3)
    model.save('modello_numeri.model')

    getter(c)

    width = 28
    height = 28

    img = Image.open(immagine_salvata)
    img = img.resize((width,height), Image.BICUBIC)
    img.save(immagine_salvata)

    img = PIL.Image.open("immagine.jpg").convert("L")
    img_ar = np.array(img)

    img_ar_norm = tf.keras.utils.normalize(img_ar, axis=1)

    plt.imshow(img_ar, cmap=plt.cm.binary)
    plt.show()  

    #make prediction
    predizione = model.predict(img_ar_norm, batch_size=None)
    risposta = np.argmax(predizione[0])

    print(risposta)

    #risultato = tk.Label(window, text=risposta, font=("Helvetica",57))
    #risultato.grid(row=0, column=1, sticky="E")

这部分给了我这些错误：

Tkinter 回调 Traceback 中的异常（最近一次调用最后一次）：文件 >“C:\Users\Massimo\AppData\Local\Programs\Python\Python37\lib\tkinter__init__.py”，第 1705 行，调用中 返回 self.func(*args) 文件“c:/Users/Massimo/AppVisualCode/disegno.py”，第 91 行，Analizza predizione = model.predict(img_ar_norm, batch_size=None) 文件“C:\Users\Massimo\ AppData\Roaming\Python\Python37\site-packages\tensorflow\python\keras\engine\training.py”，第 1113 行，在 predict self、x、batch_size=batch_size、verbose=verbose、steps=steps）文件“C： \Users\Massimo\AppData\Roaming\Python\Python37\site-packages\tensorflow\python\keras\engine\training_arrays.py”，第 329 行，在 model_iteration batch_outs = f(ins_batch) 文件“C:\Users\Massimo\ AppData\Roaming\Python\Python37\site-packages\tensorflow\python\keras\backend.py"，第 3076 行，正在调用中 run_metadata=self.run_metadata）文件“C:\Users\Massimo\AppData\Roaming\Python\Python37\site-packages\tensorflow\python\client\session.py”，第 1439 行，调用 run_metadata_ptr）文件“C:\ Users\Massimo\AppData\Roaming\Python\Python37\site-packages\tensorflow\python\framework\errors_impl.py"，第 528 行，退出 c_api.TF_GetCode(self.status.status)) tensorflow.python.framework.errors_impl .InvalidArgumentError：矩阵大小不兼容：In[0]：[28,28]，In[1]：[784,128] [[{{node MatMul}}]]

我知道预测的矩阵大小应该有问题，但为什么呢？我正在保存图像，将其大小调整为 28x28，黑白，然后像 x_train[0] 一样作为 np 数组打开

谢谢麦克斯

4

0 回答 0