Apple-2-SW
/
ii-pix
mirror of https://github.com/KrisKennaway/ii-pix.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Create data directory before writing to it

This commit is contained in:
kris 2023-01-31 21:12:42 +00:00
parent 05a3624866
commit 2ab582790c
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
precompute_conversion.py
View File

@ -6,6 +6,8 @@ colours when optimizing the image conversion (since this perceptual
difference corresponds to the Euclidean distance in this colour space) difference corresponds to the Euclidean distance in this colour space)
""" """
import os
import colour import colour
import numpy as np import numpy as np
@ -14,7 +16,15 @@ def srgb_to_linear_rgb_array(a: np.ndarray, gamma=2.4) -> np.ndarray:
return np.where(a <= 0.04045, a / 12.92, ((a + 0.055) / 1.055) ** gamma) return np.where(a <= 0.04045, a / 12.92, ((a + 0.055) / 1.055) ** gamma)
DATA_DIR = 'data'
def main(): def main():
try:
os.mkdir(DATA_DIR, mode=0o755)
except FileExistsError:
pass
print("Precomputing conversion matrix from 24-bit RGB to CAM16UCS colour " print("Precomputing conversion matrix from 24-bit RGB to CAM16UCS colour "
"space") "space")
# Compute matrix of all 24-bit RGB values, normalized to 0..1 range # Compute matrix of all 24-bit RGB values, normalized to 0..1 range
@ -30,7 +40,7 @@ def main():
rgb24_to_cam16ucs = colour.convert(all_rgb24, "RGB", "CAM16UCS").astype( rgb24_to_cam16ucs = colour.convert(all_rgb24, "RGB", "CAM16UCS").astype(
np.float32) np.float32)
del all_rgb24 del all_rgb24
np.save("data/rgb24_to_cam16ucs.npy", np.ascontiguousarray( np.save("%s/rgb24_to_cam16ucs.npy" % DATA_DIR, np.ascontiguousarray(
rgb24_to_cam16ucs)) rgb24_to_cam16ucs))
del rgb24_to_cam16ucs del rgb24_to_cam16ucs
@ -61,7 +71,7 @@ def main():
rgb12_iigs_to_cam16ucs = colour.convert( rgb12_iigs_to_cam16ucs = colour.convert(
rgb12_iigs, "RGB", "CAM16UCS").astype(np.float32) rgb12_iigs, "RGB", "CAM16UCS").astype(np.float32)
del rgb12_iigs del rgb12_iigs
np.save("data/rgb12_iigs_to_cam16ucs.npy", np.ascontiguousarray( np.save("%s/rgb12_iigs_to_cam16ucs.npy" % DATA_DIR, np.ascontiguousarray(
rgb12_iigs_to_cam16ucs)) rgb12_iigs_to_cam16ucs))
del rgb12_iigs_to_cam16ucs del rgb12_iigs_to_cam16ucs