diff options
Diffstat (limited to 'test_colorchecker.py')
-rw-r--r-- | test_colorchecker.py | 27 |
1 files changed, 19 insertions, 8 deletions
diff --git a/test_colorchecker.py b/test_colorchecker.py index 612d7fb..1503874 100644 --- a/test_colorchecker.py +++ b/test_colorchecker.py @@ -1,14 +1,25 @@ import numpy as np from colour import * -from colour.recovery import XYZ_to_sd_Jakob2019 -from gsoc_common import D65, plot_comparison +from colour.recovery import RGB_to_sd_Jakob2019 +from gsoc_common import * + # This demo goes through SDs in a color checker if __name__ == "__main__": - for name, sd in COLOURCHECKER_SDS['ColorChecker N Ohta'].items(): - XYZ = sd_to_XYZ(sd, illuminant=D65) / 100 + for name, sd in COLOURCHECKER_SDS['ColorChecker N Ohta'].items(): + XYZ = sd_to_XYZ(sd, illuminant=D65) / 100 + + RGB = XYZ_to_RGB( + XYZ, + D65_xy, + colourspace.whitepoint, + colourspace.XYZ_to_RGB_matrix, + ) + + recovered_sd, error = RGB_to_sd_Jakob2019( + RGB, + colourspace, + return_error=True + ) - print("Color checker: The target is '%s' with X=%g, Y=%g, Z=%g" - % (name, *XYZ)) - recovered_sd, error = XYZ_to_sd_Jakob2019(XYZ, return_error=True) - plot_comparison(sd, recovered_sd, name, error, D65) + plot_comparison(sd, recovered_sd, name, error, D65) |