Флойд дизеринг с KMeans - PullRequest
Новая
       25

Флойд дизеринг с KMeans

0 голосов
/ 26 апреля 2019

Я пытаюсь реализовать дизеринг Флойд-Штайнберга в Python после использования KMeans.Я понял, что после сглаживания я получаю цвета, которые не включены в уменьшенную палитру, поэтому я снова изменяю изображение с помощью KMeans.Однако при попытке с этой картинкой я не вижу никакого сглаживания вообще.Я застрял, я устал - пожалуйста, помогите мне.Мои идеи почти исчезли. 

from PIL import Image
import numpy as np
import matplotlib.pyplot as plt
from sklearn.cluster import KMeans

k = 16

im = Image.open('Image.png').convert('RGB') #Image converted to RGB
pic = np.array(im, dtype = np.float)/255 #Enables imshow()
im.close()

def kmeans(pic): #Prepares algorithmic data
    v, c, s = pic.shape
    repic = np.resize(pic, (c*v, 3))

    kme = KMeans(n_clusters = k).fit(repic)
    cl = kme.cluster_centers_

    return kme, cl, repic, v, c

kme, cl, repic, v, c = kmeans(pic)
pred = kme.predict(repic)

def picture(v, c, cl, pred): #Creates a picture with reduced colors
    image = np.ones((v, c, 3))
    ind = 0
    for i in range(v):
        for j in range(c):
            image[i][j] = cl[pred[ind]]
            ind+=1
    return image

image = picture(v, c, cl, pred)

def dither(pic, image): #Floyd-Steinberg dithering
    v, c, s = pic.shape
    Floyd = np.copy(image)
    for i in range(1, v-1):
        for j in range(1, c-1):
            quan = pic[i][j] - image[i][j]
            Floyd[i][j + 1] = quan * (np.float(7 / 16)) + pic[i][j + 1]
            Floyd[i + 1][j - 1] = quan * (np.float(5 / 16)) + pic[i + 1][j - 1]
            Floyd[i + 1][j] = quan * (np.float(3 / 16)) + pic[i + 1][j]
            Floyd[i + 1][j + 1] = quan * (np.float(1 / 16)) + pic[i + 1][j + 1]
    return Floyd

fld = dither(pic, image)
a1, a2, reim, a3, a4 = kmeans(fld)
lab = kme.predict(reim)
Floyd = picture(v, c, cl, lab)

plt.imshow(Floyd)
plt.show()
...