17 lines
518 B
Python
17 lines
518 B
Python
|
from imageClasification import imageClasification
|
||
|
|
|
||
|
|
#Training model
|
||
|
|
data = imageClasification.train_model("../SI_InteligentnyWozekWidlowy/imageClasification/images")
|
||
|
|
|
||
|
|
|
||
|
|
#image_path = ścieżka do klasyfikowanego obrazka
|
||
|
|
#data[0] = model (wytrenowany model
|
||
|
|
#data[1] = class_names
|
||
|
|
image_classified = imageClasification.image_clasification("./th-367101945.jpg",
|
||
|
|
data[0],
|
||
|
|
data[1])
|
||
|
|
|
||
|
|
|
||
|
|
print(image_classified)
|
||
|
|
|