python 图片转矩阵

最新推荐文章于 2023-03-21 15:46:39 发布

_miccretti

最新推荐文章于 2023-03-21 15:46:39 发布

阅读量5.4k

点赞数 4

CC 4.0 BY-SA版权

文章标签： python 人工智能

原文链接：https://juejin.im/post/5c3af57a6fb9a049d4420ac2

本文介绍了一种使用Python将图片转换为矩阵并反转换回图片的方法。通过使用PIL库读取图片，将其像素数据转换为浮点数矩阵，再通过matplotlib库展示图片。此外，还提供了将矩阵数据还原成原始图片的功能。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

from PIL import Image
import numpy as np
import matplotlib.pyplot as plt

def ImageToMatrix(filename):
    # 读取图片
    im = Image.open(filename)
    width,height = im.size
    data = im.getdata()
    data = np.matrix(data,dtype='float')/255
    # 这里处理的是rgb三通道的图片
    new_data = np.reshape(data.getA(),(height,width,3))
   # 单色图片：new_data = np.reshape(data,-1,height*width)
    return new_data

def MatrixToImage(data):
    # 显示图片
    data = data*255
    new_im = Image.fromarray(data.astype(np.uint8))
    return new_im

filename = '/Users/gaowenfeng/Documents/图片/头像.png'
data = ImageToMatrix(filename)
# print(data)
new_im = MatrixToImage(data)
plt.imshow(data, cmap=plt.cm.gray, interpolation='nearest')
new_im.show()
复制代码