s444463
/
projekt_widzenie
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

update paths

This commit is contained in:
Mikołaj Pokrywka 2023-01-22 17:22:30 +01:00
parent 50be2b3bac
commit d019e3f28a
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
model_pred/check_model_on_cpu/check.py
View File

@ -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)
pred_sign_char(path_to_img)

4
model_pred/pred.py
View File

@ -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]
return pred[0]