From e8b8de741d9b27dd944751545ead262bd34a4ecf Mon Sep 17 00:00:00 2001 From: Pavel Date: Sun, 4 Jun 2023 19:16:50 +0200 Subject: [PATCH] changes for better understanding of code --- AI_brain/image_recognition.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/AI_brain/image_recognition.py b/AI_brain/image_recognition.py index 8355cd1..47f766f 100644 --- a/AI_brain/image_recognition.py +++ b/AI_brain/image_recognition.py @@ -5,10 +5,11 @@ 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 +#You can download model from https://uam-my.sharepoint.com/:f:/g/personal/pavbia_st_amu_edu_pl/EmBHjnETuk5LiCZS6xk7AnIBNsnffR3Sygf8EX2bhR1w4A +#Change the path to model + to datasets (string 12 + strings 35,41,47,53) class VacuumRecognizer: - model = keras.models.load_model('AI_brain\model.h5') + model = keras.models.load_model('AI_brain\model.h5') #Neuron model path def recognize(self, image_path) -> str: class_names = ['Banana', 'Cat', 'Earings', 'Plant'] @@ -31,25 +32,25 @@ class VacuumRecognizer: 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_image_paths = 'C:\\Users\\Pavel\\Desktop\\AI\\Machine_learning_2023\\AI_brain\\Image_datasetJPGnewBnW\\check\\Plant' #Plant dataset path 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' + earnings_image_paths = 'C:\\Users\\Pavel\\Desktop\\AI\\Machine_learning_2023\\AI_brain\\Image_datasetJPGnewBnW\\check\\Earings' #Earings dataset path 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_image_paths = 'C:\\Users\\Pavel\\Desktop\\AI\\Machine_learning_2023\\AI_brain\\Image_datasetJPGnewBnW\\check\\Banana' #Banana dataset path 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_image_paths = 'C:\\Users\\Pavel\\Desktop\\AI\\Machine_learning_2023\\AI_brain\\Image_datasetJPGnewBnW\\check\\Cat' #Cat dataset path cat_dir = os.listdir(cat_image_paths)