Random walk recolor method of the reduced palette class

reduced_palette.py

class KMeansReducedPalette:
    # ... omitted

    def random_walk_recolor(self, image, max_steps):
        original_shape = image.shape
        image = self._preprocess(image)
        centroids = self.kmeans.predict(image)
        start = np.round(self.kmeans.cluster_centers_[centroids])

        diff = np.zeros(image.shape)

        for _ in range(max_steps):
            walk = np.eye(3)[np.random.randint(0, 3, size=image.shape[0])]
            sign = np.random.choice([-1, 1], size=(image.shape[0], 1))
            diff += walk * sign

        recolor = np.clip(start + diff, 0, 255).astype(np.uint8)

        return recolor.reshape(original_shape)