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