利用Python-PIL库的Image模块将RGB矩阵还原成图片
最近在用Python模拟实现k-means算法时,遇到了一个大问题,如何将聚类完成后的RGB矩阵还原为图片?
下面用一个小示例记录一下实现的过程。
步骤
1.打开图像,并获得其对应的RGB矩阵
#读取图像
from PIL import Image
def getImgData():
img = Image.open('example.jpg','r')
print(img.size)
lst = list(img.getdata())
return lst
img.size为(500,358)
2.将RGB矩阵转为n*m的列表;n指的是图片的行数(500),m指的是图片的列数(358)
lst = getImgData()
lst1 = []
for i in range(0,500):
lst1.append(lst[357*i:357*i+357])
3.验证一下是否转换成功
for val,i in zip(lst1,range(0,len(lst1))):
print(val)
print(len(val))
4.将每个RGB转为像素,然后保存成新的图片
newimg = Image.new('RGB',(500,358))
i=0
for val,i in zip(lst1,range(0,len(lst1))):
for j in range(0,len(val)):
newimg.putpixel((i,j),(int(val[j][0]),int(val[j][1]),int(val[j][2])))
newimg.save('flag.jpg')
完整代码
#读取图像
from PIL import Image
def getImgData():
img = Image.open('example.jpg','r')
print(img.size)
lst = list(img.getdata())
return lst
lst = getImgData()
lst1 = []
for i in range(0,500):
lst1.append(lst[357*i:357*i+357])
for val,i in zip(lst1,range(0,len(lst1))):
print(val)
print(len(val))
print(i)
newimg = Image.new('RGB',(500,358))
i=0
for val,i in zip(lst1,range(0,len(lst1))):
for j in range(0,len(val)):
newimg.putpixel((i,j),(int(val[j][0]),int(val[j][1]),int(val[j][2])))
newimg.save('flag.jpg')
运行结果
本人是Python菜鸟一个,请大家及时批评指正。