SI_InteligentnyWozekWidlowy/imageClasification/how_to_use.py
2022-05-27 00:03:24 +02:00

16 lines
487 B
Python

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)