SponselMa71420/Projekt_Dyschromasie
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Angepasster Daltonizationalgorithmus

Browse Source 
https://github.com/indranilsinharoy/daltonization/blob/master/daltonize.py

Implementierung des gezeigten Algorithmus. Wichtig ist die Int64 Konvertierung , um Überläufe zu verhindern.
This commit is contained in:
Max Sponsel 2020-09-17 12:23:02 +02:00
parent b78f3c2908
commit 638fadeb28
2 changed files with 12 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
Beispielbilder/ishihara_plate_45_protanopia_daltonize.jpg Normal file
View File

Binary file not shown.
Side by Side

After

Width:  |  Height:  |  Size: 19 KiB

19
Code/Farbaenderung.py
View File

@ -88,7 +88,7 @@ class Dyschromasie:
script_dir = sys.path[0] script_dir = sys.path[0]
path = script_dir[:-4] + "Beispielbilder\Fall_trees.jpg" path = script_dir[:-4] + r'Beispielbilder\Fall_trees.jpg'
image = cv2.cvtColor(cv2.imread(path), cv2.COLOR_BGR2RGB) image = cv2.cvtColor(cv2.imread(path), cv2.COLOR_BGR2RGB)
rows, cols, kanaele = image.shape rows, cols, kanaele = image.shape
@ -97,13 +97,13 @@ p = Dyschromasie(image, rows, cols, kanaele, 1, 'p')
simulated_image = p.Simulate() simulated_image = p.Simulate()
E = np.copy(simulated_image) E = np.copy(simulated_image).astype('int64')
for i in range(rows): for i in range(rows):
for j in range(cols): for j in range(cols):
for x in range(3): for x in range(3):
E[i, j, x] = abs(int(simulated_image[i, j, x]) - int(image[i, j, x])) E[i, j, x] = abs(int(image[i, j, x]) - int(simulated_image[i, j, x]))
ERR = np.zeros_like(image) ERR = np.zeros_like(image).astype('int64')
err2mod = np.array([[0,0,0],[0.7,1,0],[0.7,0,1]]) err2mod = np.array([[0,0,0],[0.7,1,0],[0.7,0,1]])
@ -112,12 +112,12 @@ for i in range(rows):
err = E[i,j,:3] err = E[i,j,:3]
ERR[i,j,:3] = np.dot(err2mod, err) ERR[i,j,:3] = np.dot(err2mod, err)
dtpn = np.copy(image) dtpn = np.copy(image).astype('int64')
for i in range(rows): for i in range(rows):
for j in range(cols): for j in range(cols):
for x in range(3): for x in range(3):
dtpn[i, j, x] = (int(ERR[i, j, x]) + int(image[i, j, x])) dtpn[i, j, x] = abs(int(ERR[i, j, x]) + int(image[i, j, x]))
for i in range(rows): for i in range(rows):
for j in range(cols): for j in range(cols):
@ -128,5 +128,10 @@ for i in range(rows):
dtpn[i, j, 2] = max(0, dtpn[i, j, 2]) dtpn[i, j, 2] = max(0, dtpn[i, j, 2])
dtpn[i, j, 2] = min(255, dtpn[i, j, 2]) dtpn[i, j, 2] = min(255, dtpn[i, j, 2])
cv2.imshow('Dalt_Img', cv2.cvtColor(dtpn, cv2.COLOR_RGB2BGR)) result = dtpn.astype('uint8')
dalt = Dyschromasie(result,rows,cols, kanaele, 1, 'p')
dalt_p = dalt.Simulate()
cv2.imshow('Dalt_Img', cv2.cvtColor(dalt_p, cv2.COLOR_RGB2BGR))
cv2.waitKey(0) cv2.waitKey(0)