kmeans图像压缩

你的名字 2023-06-24 06:23 146阅读 0赞

# -*- coding: utf-8 -*-
    
    from skimage import io
    from sklearn.cluster import KMeans
    import numpy as np
    
    image = io.imread('test2.jpg')
    io.imshow(image)
    io.show()
    
    rows = image.shape[0]
    cols = image.shape[1]
     
    # 取出图像每一个点作为样本，取出他的三个通道
    image = image.reshape(image.shape[0]*image.shape[1],3)
    # 本来0-256个可取的范围，现在压缩成0-128个簇，cpu个数，迭代次数
    kmeans = KMeans(n_clusters = 128, n_init=10, max_iter=200)
    kmeans.fit(image)
    
    clusters = np.asarray(kmeans.cluster_centers_,dtype=np.uint8) 
    labels = np.asarray(kmeans.labels_,dtype=np.uint8 )  
    labels = labels.reshape(rows,cols);
    
    print (clusters.shape)
    np.save('codebook_test.npy',clusters)    
    io.imsave('compressed_test.jpg',labels)

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MjI2MDEwMg_size_16_color_FFFFFF_t_70]

image = io.imread('compressed_test.jpg')
    io.imshow(image)
    io.show()

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MjI2MDEwMg_size_16_color_FFFFFF_t_70 1]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MjI2MDEwMg_size_16_color_FFFFFF_t_70]: https://img-blog.csdnimg.cn/20191220225911148.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MjI2MDEwMg==,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MjI2MDEwMg_size_16_color_FFFFFF_t_70 1]: https://img-blog.csdnimg.cn/20191220225951808.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MjI2MDEwMg==,size_16,color_FFFFFF,t_70