SI_InteligentnyWozekWidlowy/imageClasification/how_to_use.py

from Classificator import Classificator

#Training model
data = Classificator.train_model("../SI_InteligentnyWozekWidlowy/imageClasification/images")

#image_path = ścieżka do klasyfikowanego obrazka
#data[0] = model (wytrenowany model
#data[1] = class_names
image_classified = Classificator.image_clasification("./th-367101945.jpg",
                                                     data[0],
                                                     data[1])


print(image_classified)