Thesis-SampleAnalysis/compare_images.py

import os, sys, argparse

from PIL import Image
import ipfml.iqa.fr as fr

def main():

    parser = argparse.ArgumentParser(description="Compute .csv dataset file")

    parser.add_argument('--reference', type=str, help='Reference image')
    parser.add_argument('--reconstructed', type=str, help='Image to compare')
    parser.add_argument('--iqa', type=str, help='Image to compare', choices=['ssim', 'mse', 'rmse', 'mae', 'psnr'])
    args = parser.parse_args()

    param_reference = args.reference
    param_reconstructed = args.reconstructed
    param_iqa = args.iqa

    reference_image = Image.open(param_reference)
    reconstructed_image = Image.open(param_reconstructed)

    try:
        fr_iqa = getattr(fr, param_iqa)
    except AttributeError:
        raise NotImplementedError("FR IQA `{}` not implement `{}`".format(fr.__name__, param_iqa))

    print(fr_iqa(reference_image, reconstructed_image))


if __name__== "__main__":
    main()