From d019e3f28a0c6cca3960eb63c2bce62e312ec7b7 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Miko=C5=82aj=20Pokrywka?= Date: Sun, 22 Jan 2023 17:22:30 +0100 Subject: [PATCH] update paths --- model_pred/check_model_on_cpu/check.py | 4 ++-- model_pred/pred.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/model_pred/check_model_on_cpu/check.py b/model_pred/check_model_on_cpu/check.py index 5a48783..d34a6c0 100644 --- a/model_pred/check_model_on_cpu/check.py +++ b/model_pred/check_model_on_cpu/check.py @@ -2,7 +2,7 @@ import numpy as np import tensorflow as tf from tensorflow import keras -model = keras.models.load_model('../sign_car_detection_model') +model = keras.models.load_model('../sign_char_detection_model') class_names = ['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z', 'del', 'nothing', 'space'] @@ -24,4 +24,4 @@ def pred_sign_char(path_to_img: str): return pred[0] -pred_sign_char(path_to_img) \ No newline at end of file +pred_sign_char(path_to_img) diff --git a/model_pred/pred.py b/model_pred/pred.py index 66c0ce3..679909c 100644 --- a/model_pred/pred.py +++ b/model_pred/pred.py @@ -2,7 +2,7 @@ import numpy as np import tensorflow as tf from tensorflow import keras -model = keras.models.load_model('sign_car_detection_model') +model = keras.models.load_model('sign_char_detection_model') class_names = ['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z', 'del', 'nothing', 'space'] @@ -20,4 +20,4 @@ def pred_sign_char(path_to_img: str): print(result) pred.append(class_names[np.argmax(result)]) print(pred) - return pred[0] \ No newline at end of file + return pred[0]