mattmcw
/
marker_separation
1
0
Fork 0
Code Issues 3 Pull Requests Packages Projects Releases Wiki Activity

WIP

This commit is contained in:
Matt McWilliams 2023-10-23 00:01:31 -04:00
parent 07f9bc35e5
commit f3123e38c3
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
py/posterize.py
View File

@ -48,7 +48,7 @@ class Posterize:
def determine_colors (self): def determine_colors (self):
reshaped = self.image.reshape(-1, self.image.shape[2]) reshaped = self.image.reshape(-1, self.image.shape[2])
self.original_colors = np.unique(reshaped, axis=0) self.original_colors = np.unique(reshaped, axis=0)
print(self.original_colors) #print(self.original_colors)
for i in range(self.n_colors) : for i in range(self.n_colors) :
mask = self.extract_color(self.image, self.original_colors[i]) mask = self.extract_color(self.image, self.original_colors[i])
cv2.imwrite(f'{i}.png', mask) cv2.imwrite(f'{i}.png', mask)