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

Erster Daltonizationalgorithmus

Browse Source
This commit is contained in:
Max Sponsel 2020-09-17 11:41:00 +02:00
parent 804bc74885
commit b78f3c2908
2 changed files with 45 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
Beispielbilder/rot-gruen-schwaeche-test-bild.jpg Normal file
View File

Binary file not shown.
Side by Side

After

Width:  |  Height:  |  Size: 31 KiB

47
Code/Farbaenderung.py
View File

@ -1,7 +1,6 @@
import numpy as np
import cv2
import sys
def createGammaLookup():
return np.array([removeGammaCorrection(i) for i in np.arange(0, 256)]).astype("float64")
@ -87,3 +86,47 @@ class Dyschromasie:
return self.sim_image
script_dir = sys.path[0]
path = script_dir[:-4] + "Beispielbilder\Fall_trees.jpg"
image = cv2.cvtColor(cv2.imread(path), cv2.COLOR_BGR2RGB)
rows, cols, kanaele = image.shape
p = Dyschromasie(image, rows, cols, kanaele, 1, 'p')
simulated_image = p.Simulate()
E = np.copy(simulated_image)
for i in range(rows):
for j in range(cols):
for x in range(3):
E[i, j, x] = abs(int(simulated_image[i, j, x]) - int(image[i, j, x]))
ERR = np.zeros_like(image)
err2mod = np.array([[0,0,0],[0.7,1,0],[0.7,0,1]])
for i in range(rows):
for j in range(cols):
err = E[i,j,:3]
ERR[i,j,:3] = np.dot(err2mod, err)
dtpn = np.copy(image)
for i in range(rows):
for j in range(cols):
for x in range(3):
dtpn[i, j, x] = (int(ERR[i, j, x]) + int(image[i, j, x]))
for i in range(rows):
for j in range(cols):
dtpn[i, j, 0] = max(0, dtpn[i, j, 0])
dtpn[i, j, 0] = min(255, dtpn[i, j, 0])
dtpn[i, j, 1] = max(0, dtpn[i, j, 1])
dtpn[i, j, 1] = min(255, dtpn[i, j, 1])
dtpn[i, j, 2] = max(0, dtpn[i, j, 2])
dtpn[i, j, 2] = min(255, dtpn[i, j, 2])
cv2.imshow('Dalt_Img', cv2.cvtColor(dtpn, cv2.COLOR_RGB2BGR))
cv2.waitKey(0)