Improve ssim example

This commit is contained in:
Johannes Schönberger
2013-11-30 19:39:39 +01:00
parent 64b9c1847b
commit 1720629670
+5 -1
View File
@@ -22,12 +22,16 @@ but with very different mean structural similarity indices.
"""
import numpy as np
import matplotlib
import matplotlib.pyplot as plt
from skimage import data, img_as_float
from skimage.measure import structural_similarity as ssim
matplotlib.rcParams['font.size'] = 9
img = img_as_float(data.camera())
rows, cols = img.shape
@@ -41,7 +45,7 @@ def mse(x, y):
img_noise = img + noise
img_const = img + abs(noise)
f, (ax0, ax1, ax2) = plt.subplots(1, 3)
f, (ax0, ax1, ax2) = plt.subplots(nrows=1, ncols=3, figsize=(8, 4))
mse_none = mse(img, img)
ssim_none = ssim(img, img, dynamic_range=img.max() - img.min())