Added script for similarity metric calculation.

scripts/similarity.py

@@ -0,0 +1,48 @@
+#!/usr/bin/python3
+
+# BSD 2-Clause License
+# 
+# Copyright (c) 2021, Christoph Neuhauser
+# All rights reserved.
+# 
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions are met:
+# 
+# 1. Redistributions of source code must retain the above copyright notice, this
+#    list of conditions and the following disclaimer.
+# 
+# 2. Redistributions in binary form must reproduce the above copyright notice,
+#    this list of conditions and the following disclaimer in the documentation
+#    and/or other materials provided with the distribution.
+# 
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+# DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+# SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+# CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+# OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+
+import sys
+import math
+import skimage
+import skimage.io
+import skimage.metrics
+
+
+if __name__ == '__main__':
+    filename_gt = sys.argv[1]
+    filename_approx = sys.argv[2]
+    img_gt = skimage.io.imread(filename_gt)
+    img_approx = skimage.io.imread(filename_approx)
+    mse = skimage.metrics.mean_squared_error(img_gt, img_approx)
+    psnr = skimage.metrics.peak_signal_noise_ratio(img_gt, img_approx)
+    data_range=img_gt.max() - img_approx.min()
+    ssim = skimage.metrics.structural_similarity(img_gt, img_approx, data_range=data_range, multichannel=True)
+    print(f'MSE: {mse}')
+    print(f'RMSE: {math.sqrt(mse)}')
+    print(f'PSNR: {psnr}')
+    print(f'SSIM: {ssim}')