Zaktualizuj 'image recognition.md'

This commit is contained in:
Karol Idaszak 2020-06-15 13:26:29 +00:00
parent d69423958f
commit 283acf21f1

View File

@ -1 +1,33 @@
# Rozpoznawanie obrazu wykorzystująć konwolucyjną sieć neuronową ## Rozpoznawanie obrazu wykorzystująć konwolucyjną sieć neuronową
## Opis
Konwolucyjna sieć neuronowa wyuczona na 10000 zdjęć o wymiarze 64x64x3 służąca do rozpoznawania zdjeć dań.
## Dane
Rodzaje dań dostępnych do zamówienia
```
CATEGORIES = [
"apple_pie",
"club_sandwich",
"greek_salad",
"hamburger",
"hot_dog",
"ice_cream",
"lasagna",
"pizza",
"steak",
"waffles"
]
```
## Implementacja
Zostaje wybrane losowo zdjęcie, następnie na podstawie wyuczonego modelu sieci neuronowej następuje rozpoznanie potrawy. Jeśli potrawa jest tą którą zamówił klient następuje przerwanie pętli.
```
def image_recognition():
for _ in range(100):
photo = random.choice(menu)
prediction = model.predict(np.expand_dims(photo[0], axis=0))
max_value = prediction[0].max()
idx = np.where(prediction[0]==max_value)
if CATEGORIES[idx[0][0]] == waiter.order_list[-1]:
break
waiter.order_list.pop()
```