提问人:Uwe_98 提问时间:10/2/2019 最后编辑:Uwe_98 更新时间:10/9/2019 访问量:1639
使用 Python 3.6 将二维 rgb 矩阵转换为 PNG 图像
Conversion 2-dimensional rgb matrix to PNG Image with Python 3.6
问:
我需要使用 Python 3.6 转换 24 维 3.6 位 rgb- 矩阵
[[rr11gg11bb11, rr12gg12bb12,.. ..,rr1mgg1mbb1m],
.
.
[rri1gg11bb11, rr12ggi2bb12,.. ..,rr1mgg1mbb1m]
.
.
[rrn1ggn1bbn1, rrn2ggn2bbn2,.. ..,rrmnggmnbbmn]]
转换为 TIF、PNG 和 JPG 格式的图像。
我还得到了文件形式的矩阵
image.bin.
从这个二进制文件中,我可以借助 RAW 像素查看器对图像进行编辑:
http://rawpixels.net/
我找到了 PIL 库,但我没有找到任何从这些数据构建图像对象的方法。我该怎么做?我使用 Python 3.6。
现在我尝试了以下方法:
im = Image.frombuffer('RGB', (self.width, self.height), self.matrix, 'raw',
'RGB', 0, 1)
logger.info("image object created.")
im.save('result.png')
logger.info("image saved as result.png")
但我得到的信息是:
TypeError: a bytes-like object is required, not 'list'
现在我发现我必须将对象转换为字节数组:
npm = np.array(self.matrix)
arr = bytearray(npm)
print(arr)
im = Image.frombuffer('RGB', (self.width, self.height), arr, 'raw',
'RGB', 0, 1)
当我使用“L”或“RGBA”时,转换会起作用,但不能使用“RGB”。奇怪的是,在这种情况下,它拒绝字节数组对象。
答:
更新的答案
因此,从本质上讲,您没有 Numpy 数组,但您有一个列表列表,其中每个元素都是一个 24 位数字,代表一个 RGB 三元组 RGB888。
所以,我可以像这样表示你的列表:
f=[
[0xFFFFFF,0xFFFFFF,0xFFFFFF,0xFFFFFF,0x000000,0xFFFFFF,0xFFFFFF,0xFFFFFF,0xFFFFFF,0xFFFFFF],
[0xFFFFFF,0xFFFFFF,0xFFFFFF,0x000000,0x000000,0xFFFFFF,0xFFFFFF,0xFFFFFF,0xFFFFFF,0xFFFFFF],
[0xFFFFFF,0xFFFFFF,0x000000,0xFFFFFF,0x000000,0xFFFFFF,0xFFFFFF,0xFFFFFF,0xFFFFFF,0xFFFFFF],
[0xFFFFFF,0xFFFFFF,0xFFFFFF,0xFFFFFF,0x000000,0xFFFFFF,0xFFFFFF,0xFFFFFF,0xFFFFFF,0xFFFFFF],
[0xFFFFFF,0xFFFFFF,0xFFFFFF,0xFFFFFF,0x000000,0xFFFFFF,0xFFFFFF,0xFFFFFF,0xFFFFFF,0xFFFFFF]
]
因此,我可以将其转换为 uint32 元素的 Numpy 数组,如下所示:
na = np.array(f).astype(np.uint32)
然后制作一个 PIL 喜欢使用的 RGB 数组:
h,w = 5,10
RGB=np.zeros((h,w,3),dtype=np.uint8)
然后我只需要将三胞胎移位并复制到正确的位置:
RGB[:,:,0] = na>>16 # Take red from top
RGB[:,:,1] = (na>>8) & 0xff # Take green from middle
RGB[:,:,2] = na & 0xff # Take blue from bottom
现在,我可以从该RGB阵列制作PIL图像并将其保存到磁盘:
pIm = Image.fromarray(RGB).save('result.png')
原始答案
让我们使用 ImageMagick 创建一个测试数据文件。首先作为 PNG,这样你就可以看到它:
magick -size 640x480 gradient:magenta-yellow image.png
现在又是同样的事情,但这次是RGB888来匹配你的文件:
magick -size 640x480 gradient:magenta-yellow -depth 8 rgb:image.bin
现在:
from PIL import Image
# Open the file and read contents into "data"
with open('image.bin','rb') as f:
data = f.read()
# Convert that into PIL Image
im = Image.frombuffer('RGB', (640, 480), data, 'raw', 'RGB', 0, 1)
# Save
im.save('result.png')
同样,您可以让 Numpy 读取文件,然后将结果转换为 PIL 图像:
import numpy as np
im = Image.fromarray(np.fromfile('image.bin',dtype=np.uint8).reshape(480,640,3))
如果你想检查你的二进制文件是否真的包含你认为它包含的内容,你也可以用ImageMagick将其转换为PNG来测试它。假设您认为文件是 640 像素宽 x 480 像素高和 RGB888,您可以要求 ImageMagick 将其转换为 PNG:
magick -depth 8 -size 640x480 RGB:<YOURFILENAME> image.png
或者,如果您认为它是具有 128 字节标头的单个 16 位单通道灰度图像,则要忽略该图像:
magick -depth 16 -size 640x480+128 GRAY:<YOURFILENAME> image.png
关键词:原始、原始图像、二进制文件、图像、图像处理、Python、ImageMagick、转换。
评论
'RGB'
'L'
ba