summaryrefslogtreecommitdiff
path: root/test_colorchecker.py
blob: 15038745619cbe9db2469e81ce86eec0ba71564c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
import numpy as np
from colour import *
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

        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
        )

        plot_comparison(sd, recovered_sd, name, error, D65)