WIP
This commit is contained in:
parent
07f9bc35e5
commit
f3123e38c3
|
@ -48,7 +48,7 @@ class Posterize:
|
|||
def determine_colors (self):
|
||||
reshaped = self.image.reshape(-1, self.image.shape[2])
|
||||
self.original_colors = np.unique(reshaped, axis=0)
|
||||
print(self.original_colors)
|
||||
#print(self.original_colors)
|
||||
for i in range(self.n_colors) :
|
||||
mask = self.extract_color(self.image, self.original_colors[i])
|
||||
cv2.imwrite(f'{i}.png', mask)
|
||||
|
|
Loading…
Reference in New Issue