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

Daltonization in die Klasse eingefügt

Browse Source 
Funktion der Daltonization in die Klasse Dyschromasie eingefügt.
This commit is contained in:
Max Sponsel 2020-09-18 10:58:28 +02:00
parent 638fadeb28
commit 8498500c68
1 changed files with 44 additions and 43 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
Code/Farbaenderung.py
View File

@ -2,6 +2,7 @@ import numpy as np
import cv2 import cv2
import sys import sys
def createGammaLookup(): def createGammaLookup():
return np.array([removeGammaCorrection(i) for i in np.arange(0, 256)]).astype("float64") return np.array([removeGammaCorrection(i) for i in np.arange(0, 256)]).astype("float64")
@ -86,36 +87,28 @@ class Dyschromasie:
return self.sim_image return self.sim_image
def daltonize(self):
script_dir = sys.path[0]
path = script_dir[:-4] + r'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).astype('int64')
for i in range(rows):
for j in range(cols):
for x in range(3):
E[i, j, x] = abs(int(image[i, j, x]) - int(simulated_image[i, j, x]))
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]])
simulated_image = self.Simulate()
for i in range(rows): E = np.copy(simulated_image).astype('int32')
for j in range(cols): for i in range(self.rows):
for j in range(self.cols):
for x in range(3):
E[i, j, x] = abs(int(self.img_mat[i, j, x]) - int(simulated_image[i, j, x]))
ERR = np.zeros_like(image).astype('int32')
for i in range(self.rows):
for j in range(self.cols):
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).astype('int64') dtpn = np.copy(image).astype('int32')
for i in range(rows): for i in range(self.rows):
for j in range(cols): for j in range(self.cols):
for x in range(3): for x in range(3):
dtpn[i, j, x] = abs(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]))
@ -130,8 +123,16 @@ for i in range(rows):
result = dtpn.astype('uint8') result = dtpn.astype('uint8')
dalt = Dyschromasie(result,rows,cols, kanaele, 1, 'p') dalt = Dyschromasie(result, self.rows, self.cols, self.kanaele, self.sim_faktor, self.sim_kind)
dalt_p = dalt.Simulate() return dalt.Simulate()
cv2.imshow('Dalt_Img', cv2.cvtColor(dalt_p, cv2.COLOR_RGB2BGR))
script_dir = sys.path[0]
path = script_dir[:-4] + r'Beispielbilder\rot-gruen-schwaeche-test-bild.jpg'
image = cv2.cvtColor(cv2.imread(path), cv2.COLOR_BGR2RGB)
rows, cols, kanaele = image.shape
dalt = Dyschromasie(image, rows, cols, kanaele, 1, 'p')
cv2.imshow('Dalt_Img', cv2.cvtColor(dalt.daltonize(), cv2.COLOR_RGB2BGR))
cv2.waitKey(0) cv2.waitKey(0)