diff options
Diffstat (limited to 'test_colorchecker.py')
-rw-r--r-- | test_colorchecker.py | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/test_colorchecker.py b/test_colorchecker.py index c934518..0368058 100644 --- a/test_colorchecker.py +++ b/test_colorchecker.py @@ -2,7 +2,7 @@ import numpy as np from colour import * from colour.plotting import * from matplotlib import pyplot as plt -from jakob_hanika import jakob_hanika +from jakob_hanika import jakob_hanika, model_sd @@ -13,7 +13,7 @@ ill_sd = SpectralDistribution(ILLUMINANTS_SDS[illuminant]) # Makes a comparison plot with SDs and swatches -def plot_comparison(target, matched_sd, label, error, ill_sd): +def plot_comparison(target, matched_sd, label, error, ill_sd, show=True): if type(target) is SpectralDistribution: target_XYZ = sd_to_XYZ(target, illuminant=ill_sd) / 100 else: @@ -33,7 +33,8 @@ def plot_comparison(target, matched_sd, label, error, ill_sd): axes = plt.subplot(2, 1, 2) plt.title("ΔE = %g" % error) - plot_multi_colour_swatches([target_swatch, matched_swatch], axes=axes) + plot_multi_colour_swatches([target_swatch, matched_swatch], + standalone=show, axes=axes) if __name__ == "__main__": # This demo goes through SDs in a color checker @@ -42,6 +43,7 @@ if __name__ == "__main__": target = XYZ_to_Lab(XYZ, ill_xy) print("Color checker: The target is '%s' with L=%g, a=%g, b=%g" % (name, *target)) - _, matched_sd, error = jakob_hanika(target, ill_sd, ill_xy) + ccp, error = jakob_hanika(target, ill_sd, ill_xy) + matched_sd = model_sd(ccp) plot_comparison(sd, matched_sd, name, error, ill_sd) |