From 283acf21f127dfe437960ac241e27e34562b7a68 Mon Sep 17 00:00:00 2001 From: Karol Idaszak Date: Mon, 15 Jun 2020 13:26:29 +0000 Subject: [PATCH] Zaktualizuj 'image recognition.md' --- image recognition.md | 34 +++++++++++++++++++++++++++++++++- 1 file changed, 33 insertions(+), 1 deletion(-) diff --git a/image recognition.md b/image recognition.md index 0d1487c..bb94ee8 100644 --- a/image recognition.md +++ b/image recognition.md @@ -1 +1,33 @@ -# Rozpoznawanie obrazu wykorzystująć konwolucyjną sieć neuronową \ No newline at end of file +## 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() +```