Реализация Флойда-Стейнберга Python - PullRequest
Новая
       26

Реализация Флойда-Стейнберга Python

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

Я использую сглаживание Флойда-Стейнберга, чтобы рассеять ошибку квантования после обработки изображения с помощью KMeans от scipy. Данные представлены в виде файла RGB - как для оттенков серого, так и для цвета. Проблема заключается в визуализации - я не замираю. 

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

im = Image.open('file.png').convert('RGB')
pic = np.array(im, dtype = np.float)/255
im.close()

Я бы хотел опустить часть KMeans и сосредоточиться на Флойд-Штайнберге: 

"""pic - as above, original array; image - processed image"""

def dither(pic, image):
    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] #Quantization error
            Floyd[i][j + 1] = quan * (np.float(7 / 16)) + Floyd[i][j + 1]
            Floyd[i + 1][j - 1] = quan * (np.float(3 / 16)) + Floyd[i + 1][j - 1]
            Floyd[i + 1][j] = quan * (np.float(5 / 16)) + Floyd[i + 1][j]
            Floyd[i + 1][j + 1] = quan * (np.float(1 / 16)) + Floyd[i + 1][j + 1]
    return Floyd

Floyd = dither(pic, image)

plt.imshow(Floyd)
plt.show()

Я получаю небольшое смущение, когда заменяю Флойд на рис, т. Е. Floyd[i + 1][j] = quan * (np.float(5 / 16)) + pic[i + 1][j]. Тем не менее, это неправильный код! Кроме того, мне приходится иметь дело с цветами вне кластеров, поэтому я снова оцениваю новые пиксели для кластеров. Как я могу заставить это работать? Где ЭТА решающая ошибка?

...