SI_InteligentnyWozekWidlowy/imageClasification/usedAssets/how_to_use.py

16 lines
487 B
Python
Raw Normal View History

2022-05-27 00:03:24 +02:00
from Classificator import Classificator
#Training model
2022-05-27 00:03:24 +02:00
data = Classificator.train_model("../SI_InteligentnyWozekWidlowy/imageClasification/images")
#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])
print(image_classified)