mirror of
https://github.com/KrisKennaway/ii-pix.git
synced 2024-06-12 16:29:28 +00:00
Write out mmapp'ed numpy array instead of pickling
This commit is contained in:
parent
ebd9f94d98
commit
02393381df
|
@ -1,6 +1,3 @@
|
||||||
import bz2
|
|
||||||
import pickle
|
|
||||||
|
|
||||||
import dither
|
import dither
|
||||||
import colour.difference
|
import colour.difference
|
||||||
import numpy as np
|
import numpy as np
|
||||||
|
@ -28,11 +25,10 @@ def nearest_colours():
|
||||||
diffs[:, i] = colour.difference.delta_E_CIE2000(all_lab, palette_lab)
|
diffs[:, i] = colour.difference.delta_E_CIE2000(all_lab, palette_lab)
|
||||||
|
|
||||||
norm = np.max(diffs)
|
norm = np.max(diffs)
|
||||||
print(norm)
|
|
||||||
return (diffs / norm * 255).astype(np.uint8)
|
return (diffs / norm * 255).astype(np.uint8)
|
||||||
#return diffs
|
|
||||||
|
|
||||||
|
|
||||||
n = nearest_colours()
|
n = nearest_colours()
|
||||||
with bz2.open("nearest2.pickle.bz2", "wb") as f:
|
out = np.memmap(filename="distances.npy", mode="w+", dtype=np.uint8,
|
||||||
pickle.dump(n, f)
|
shape=n.shape)
|
||||||
|
out[:] = n[:]
|
||||||
|
|
Loading…
Reference in New Issue
Block a user