如何将 .txt 文件中的 RGB 值转换为在 Python 中显示图像

Question

我有一个包含 RGB 值的 .txt 文件，当我打开并读取文件时，像素值是 str 格式。如何转换这些值以在 python 中显示图像。 image.

这是我尝试读取值时的。它们都是字符串格式。

编辑：你可以在这里找到文件的链接https://drive.google.com/file/d/1mAxlcMj_SVeK0axJhbPJqO4k_egJoYli/view?usp=sharing

Answer 1

这很简单：

#!/usr/bin/env python3

import re
import numpy as np
from PIL import Image
from pathlib import Path

# Open image file, slurp the lot
contents = Path('image.txt').read_text()

# Make a list of anything that looks like numbers using a regex...
# ... taking first as height, second as width and remainder as pixels
h, w, *pixels = re.findall(r'[0-9]+', contents)

# Now make pixels into Numpy array of uint8 and reshape to correct height, width and depth
na = np.array(pixels, dtype=np.uint8).reshape((int(h),int(w),3))

# Now make the Numpy array into a PIL Image and save
Image.fromarray(na).save("result.png")

---

如果您想使用 OpenCV 而不是 PIL/Pillow 编写输出图像，请将上面的最后一行更改为以下内容，以便进行 RGB->BGR 重新排序并改用cv2.imwrite()：

# Save with OpenCV instead
cv2.imwrite('result.png', na[...,::-1])

---

如果你想写一个 PPM 文件（兼容 Photoshop、GIMP、OpenCV、PIL/Pillow 和 ImageMagick)，没有使用 PIL/Pillow 或 OpenCV 或任何额外的库，并且大约 1 /4 原始文件的大小，您可以通过将上面的最后一行替换为以下内容非常简单地以二进制形式编写它：

# Save "na" as binary PPM image
with open('result.ppm','wb') as f:
   f.write(f'P6\n{w} {h}\n255\n'.encode())
   f.write(na.tobytes())

---

其实你不需要任何Python，直接在Terminal的命令行下写一个Photoshop可以读取的NetPBM文件就可以了， GIMP、PIL/枕头

awk 'NR==1{$0="P3\n" $2 " " $1 "\n255"} {gsub(/,/,"\n")} 1' image.txt > result.ppm

那个脚本基本上是“massages”你的第一行，所以它是从这个开始的：

418 870
... rest of your data ...

到这里：

P3
870 418
255
... rest of your data ...

Answer 2

有必要确定两个步骤来解决这个问题：

1.解析文件以获取像素以及图像的宽度和高度。在这一步中，您需要知道信息是如何 保存在文件中。

2。使用 OpenCV 显示图像。下面是一个将矩阵显示为图像的基本示例：

import numpy as np
import cv2
# image of 100 x 100 pixels , with 3 channels 
height=100 
width=100
channels=3 
color_bg=(0,0,0) 
imgdim = (height, width, channels ) 
blank_image = np.full(imgdim, color_bg, np.uint8)
#Simple way to change the pixel(x=1,y=2) color to the (B=255,G=0,R=0) tuple color 
blank_image[1,2]= (255,0,0) 

cv2.imshow("blank", blank_image)
cv2.waitKey(0)

请阅读Basics operations with opencv-python 的文档以了解有关 numpy 的概念和性能建议。

Answer 3

读取txt文件需要这段代码

file= open("filename.txt")
lines = file.readlines()

然后你可以用函数获取每一行的数字

lines[0].split(",")

终于可以用枕头制作image with the data

    from PIL import Image
     
    img = Image.new('RGB', (60, 30), color = 'red')
    img.save('pil_red.png')