import sys
from ImageLoader import ImageLoader
from ClusteringHandler import ClusteringHandler
from HTMLCreator import HTMLCreator
from ClustersPrinter import ClustersPrinter
import constants as c
if len(sys.argv) < 2:
print("Please pass a path to a file containing image paths list")
else:
paths_file = sys.argv[1]
images = ImageLoader.load_images(paths_file)
clustered_images = ClusteringHandler.cluster(images)
HTMLCreator.create_html(c.HTML_TEMPLATE,
c.HTML_OUTPUT,
clustered_images=clustered_images,
clusters_count=len(clustered_images))
ClustersPrinter.print_clusters_to_file(clustered_images, c.FILE_OUTPUT)
def load_image_dataset(image_dir) -> Tuple[pd.DataFrame, pd.DataFrame, pd.Series, pd.Series]:
loader = ImageLoader(image_dir)
x_train, x_test, y_train, y_test = loader.load_images()
x_train = loader.scale_data(x_train)
x_test = loader.scale_data(x_test)
return x_train, x_test, y_train, y_test
