evaluation.py

from ASCII_functions import generate_ascii_art, calculate_similarity, process_image
from PIL import Image
import argparse

if __name__ == "__main__":
    parser = argparse.ArgumentParser(description="Convert an image to ASCII art and evaluate its similarity.")
    parser.add_argument("image_path", type=str, help="Path to the input image file")

    args = parser.parse_args()

    image_path = args.image_path

    original_image = Image.open(image_path)

    ascii_art = generate_ascii_art(original_image)

    adjusted_image = process_image(original_image)
    adjusted_image.show()

    print(ascii_art)

    ssim_value, mse = calculate_similarity(original_image, ascii_art)

    print(f"SSIM: {ssim_value:.4f}")
    print(f"MSE: {mse:.4f}")