2022-05-27 00:03:24 +02:00
|
|
|
from Classificator import Classificator
|
2022-05-25 21:16:09 +02:00
|
|
|
|
|
|
|
#Training model
|
2022-05-27 00:03:24 +02:00
|
|
|
data = Classificator.train_model("../SI_InteligentnyWozekWidlowy/imageClasification/images")
|
2022-05-25 21:16:09 +02:00
|
|
|
|
|
|
|
#image_path = ścieżka do klasyfikowanego obrazka
|
|
|
|
#data[0] = model (wytrenowany model
|
|
|
|
#data[1] = class_names
|
2022-05-27 00:03:24 +02:00
|
|
|
image_classified = Classificator.image_clasification("./th-367101945.jpg",
|
|
|
|
data[0],
|
|
|
|
data[1])
|
2022-05-25 21:16:09 +02:00
|
|
|
|
|
|
|
|
|
|
|
print(image_classified)
|
|
|
|
|