• Najnowsze pytania
  • Bez odpowiedzi
  • Zadaj pytanie
  • Kategorie
  • Tagi
  • Zdobyte punkty
  • Ekipa ninja
  • IRC
  • FAQ
  • Regulamin
  • Książki warte uwagi

Model TensorFlow pokazuje złe wyniki

Object Storage Arubacloud
0 głosów
167 wizyt
pytanie zadane 18 maja 2021 w Python przez Patryk01 Obywatel (1,270 p.)

A więc niedawno zrobiłem model oparty na danych mnist z biblioteki tensorflow (obrazki 28x28 odręcznie napisanych cyfr). Testowanie wyszło całkiem ładnie, ale gdy chciałem dodać własny obrazek napisanej cyfry, model kompletnie nie potrafił odgadnąć co to za cyfra (dla 7 zwracał np. 3). Testowałem to na różnych własnych obrazkach i niczego to nie zmieniło, wciąż nie wiem gdzie tkwi problem. Oto kod:

import tensorflow as tf
import cv2
import numpy as np


# Loading a dataset
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)
x_test = tf.keras.utils.normalize(x_test, axis=1)

# Creating a model
model = tf.keras.models.Sequential()

# Input Layer
model.add(tf.keras.layers.Flatten())

# Hidden Layers
model.add(tf.keras.layers.Dense(128, activation=tf.nn.relu))
model.add(tf.keras.layers.Dense(128, activation=tf.nn.relu))

# Output Layer
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)


img_array = cv2.imread("image.jpg", cv2.IMREAD_GRAYSCALE)
image = cv2.resize(img_array, (28, 28))

image = tf.convert_to_tensor(image)

X = np.array(image)
X = X/255.0
X = tf.expand_dims(X, [0])

prediction = model.predict(X)
print(f"prediction: {np.argmax(prediction)}")

Zaznaczyłem część kodu, której najbardziej nie jestem pewien.

Do tego jeden z własnych obrazków jakich użyłem: 
Wiem, że program nie będzie odgadywał mi wszystkiego, ale tutaj nie odgaduje kompletnie niczego. Co zrobiłem źle lub czego nie zauważyłem?

Zaloguj lub zarejestruj się, aby odpowiedzieć na to pytanie.

Podobne pytania

0 głosów
0 odpowiedzi 93 wizyt
0 głosów
1 odpowiedź 165 wizyt
pytanie zadane 6 kwietnia 2020 w Python przez bbbbbbbb Użytkownik (950 p.)
0 głosów
1 odpowiedź 231 wizyt
pytanie zadane 22 marca 2019 w Python przez Maciej Złotorowicz Gaduła (4,230 p.)

92,565 zapytań

141,417 odpowiedzi

319,602 komentarzy

61,950 pasjonatów

Motyw:

Akcja Pajacyk

Pajacyk od wielu lat dożywia dzieci. Pomóż klikając w zielony brzuszek na stronie. Dziękujemy! ♡

Oto polecana książka warta uwagi.
Pełną listę książek znajdziesz tutaj.

Akademia Sekuraka

Kolejna edycja największej imprezy hakerskiej w Polsce, czyli Mega Sekurak Hacking Party odbędzie się już 20 maja 2024r. Z tej okazji mamy dla Was kod: pasjamshp - jeżeli wpiszecie go w koszyku, to wówczas otrzymacie 40% zniżki na bilet w wersji standard!

Więcej informacji na temat imprezy znajdziecie tutaj. Dziękujemy ekipie Sekuraka za taką fajną zniżkę dla wszystkich Pasjonatów!

Akademia Sekuraka

Niedawno wystartował dodruk tej świetnej, rozchwytywanej książki (około 940 stron). Mamy dla Was kod: pasja (wpiszcie go w koszyku), dzięki któremu otrzymujemy 10% zniżki - dziękujemy zaprzyjaźnionej ekipie Sekuraka za taki bonus dla Pasjonatów! Książka to pierwszy tom z serii o ITsec, który łagodnie wprowadzi w świat bezpieczeństwa IT każdą osobę - warto, polecamy!

...