FilesExpand file tree

master

/

sepia.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

50 lines (39 loc) · 1.39 KB

master

/

sepia.py

Copy path

Top

File metadata and controls

• Code
• Blame

50 lines (39 loc) · 1.39 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

"""

Implemented an algorithm using opencv to tone an image with sepia technique

"""

from cv2 import destroyAllWindows, imread, imshow, waitKey

def make_sepia(img, factor: int):

"""

Function create sepia tone.

Source: https://en.wikipedia.org/wiki/Sepia_(color)

"""

pixel_h, pixel_v = img.shape[0], img.shape[1]

def to_grayscale(blue, green, red):

"""

Helper function to create pixel's greyscale representation

Src: https://pl.wikipedia.org/wiki/YUV

"""

return 0.2126 * red + 0.587 * green + 0.114 * blue

def normalize(value):

"""Helper function to normalize R/G/B value -> return 255 if value > 255"""

return min(value, 255)

for i in range(pixel_h):

for j in range(pixel_v):

greyscale = int(to_grayscale(*img[i][j]))

img[i][j] = [

normalize(greyscale),

normalize(greyscale + factor),

normalize(greyscale + 2 * factor),

]

return img

if __name__ == "__main__":

# read original image

images = {

percentage: imread("image_data/lena.jpg", 1) for percentage in (10, 20, 30, 40)

}

for percentage, img in images.items():

make_sepia(img, percentage)

for percentage, img in images.items():

imshow(f"Original image with sepia (factor: {percentage})", img)

waitKey(0)

destroyAllWindows()