From f3e6d04eb413557ce2780e110f64e814e1e61f25 Mon Sep 17 00:00:00 2001 From: Pavel Date: Sun, 4 Jun 2023 19:02:37 +0200 Subject: [PATCH] recognition file changes + link for neuron model --- AI_brain/image_recognition.py | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) diff --git a/AI_brain/image_recognition.py b/AI_brain/image_recognition.py index d5c097b..8355cd1 100644 --- a/AI_brain/image_recognition.py +++ b/AI_brain/image_recognition.py @@ -5,6 +5,7 @@ from tensorflow import keras import cv2 import random +#YOU CAN DOWNLOAD MODEL FROM https://uam-my.sharepoint.com/:f:/g/personal/pavbia_st_amu_edu_pl/EmBHjnETuk5LiCZS6xk7AnIBNsnffR3Sygf8EX2bhR1w4A class VacuumRecognizer: model = keras.models.load_model('AI_brain\model.h5') @@ -18,6 +19,7 @@ class VacuumRecognizer: predictions = self.model.predict(img)[0].tolist() + # More information about neuron decision # print(img.shape) # cv2.imshow("test_show", img) # print(class_names) @@ -27,6 +29,28 @@ class VacuumRecognizer: return class_names[predictions.index(max(predictions))] + def get_random_dir(self, type) -> str: + if type == 'Plant': + plant_image_paths = 'C:\\Users\\Pavel\\Desktop\\AI\\Machine_learning_2023\\AI_brain\\Image_datasetJPGnewBnW\\check\\Plant' + plant_dirs = os.listdir(plant_image_paths) + full_path = plant_image_paths + '\\' + plant_dirs[random.randint(0, len(plant_dirs)-1)] + print(full_path) + return full_path + elif type == 'Earings': + earnings_image_paths = 'C:\\Users\\Pavel\\Desktop\\AI\\Machine_learning_2023\\AI_brain\\Image_datasetJPGnewBnW\\check\\Earings' + earning_dirs = os.listdir(earnings_image_paths) + full_path = earnings_image_paths + '\\' + earning_dirs[random.randint(0, len(earning_dirs)-1)] + print(full_path) + return full_path + elif type == 'Banana': + banana_image_paths = 'C:\\Users\\Pavel\\Desktop\\AI\\Machine_learning_2023\\AI_brain\\Image_datasetJPGnewBnW\\check\\Banana' + banana_dirs = os.listdir(banana_image_paths) + full_path = banana_image_paths + '\\' + banana_dirs[random.randint(0, len(banana_dirs)-1)] + print(full_path) + return full_path + elif type == 'Cat': + cat_image_paths = 'C:\\Users\\Pavel\\Desktop\\AI\\Machine_learning_2023\\AI_brain\\Image_datasetJPGnewBnW\\check\\Cat' + cat_dir = os.listdir(cat_image_paths) #For testing the neuron model