-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy paththinslide.py
104 lines (77 loc) · 3.07 KB
/
thinslide.py
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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
from PIL import Image
import os
testslide0 = os.getcwd() + f"{os.sep}thinslides{os.sep}0.jpg"
testslide1 = os.getcwd() + f"{os.sep}thinslides{os.sep}1.jpg"
class ColorRelation:
def __init__(self, color):
self.color = color
self.related = []
for shade_index in range(0, 3):
for i in range(1, 5):
newcolor = [*self.color]
newcolor_dn = [*self.color]
newcolor[shade_index] += i
newcolor_dn[shade_index] -= i
self.related.append(tuple(newcolor))
self.related.append(tuple(newcolor_dn))
class ColorOccurence:
def __init__(self, color, occurence=0):
self.color = color
self.occurence = occurence
class ColorFrequency:
def __init__(self, color):
self.color = color
self.frequency = 0
def increment_if_color(self, color):
if color == self.color:
self.frequency += 1
class ImagePixels:
def __init__(self, filepath:str):
img = Image.open(filepath)
self.width, self.height = img.size
# Get the pixel data
_pixels = list(img.getdata())
# If the image has multiple channels (e.g., RGB), flatten the list
if img.mode in ('RGB', 'RGBA'):
_pixels = [pixel[:3] for pixel in _pixels]
self.pixels = [px for px in _pixels if px!=_pixels[0]]
self.palette = list(set(self.pixels))
self.num_pixels = len(self.pixels)
self.num_colors = len(self.palette)
self.color_frequency = {}
for px in self.pixels:
try:
self.color_frequency[px]+=1
except KeyError:
self.color_frequency[px] = 1
self.abundant_colors = [name for name, value in self.color_frequency.items() if value > 5]
self.color_occurrences = [ColorOccurence(name, value) for name, value in self.color_frequency.items()]
self.num_abundant_colors = len(self.abundant_colors)
self.related_colors = [ColorRelation(px) for px in self.pixels]]
def print_abundant_colors(self):
for px in self.abundant_colors:
print(px)
print(self.num_abundant_colors)
def print_frequency(self):
for name, value in self.color_frequency.items():
print(f"color: {name} \t count: {value}")
def print(self):
for px in self.pixels:
print(px)
def print_palette(self):
for px in self.palette:
print(px)
def print_info(self):
print(f"pixel count: {self.num_pixels}\npalette count: {self.num_colors}\nimage width: {self.width}\nimage height: {self.height}")
def relate_colors(self):
retv = []
for rc in self.related_colors
def image_to_pixel_list(filepath: str) -> list:
# Open the image
img = Image.open(filepath)
# Get the pixel data
pixel_data = list(img.getdata())
# If the image has multiple channels (e.g., RGB), flatten the list
if img.mode in ('RGB', 'RGBA'):
pixel_data = [pixel[:3] for pixel in pixel_data]
return pixel_data