如何在Google Colab中基于R、G、B三个数值矩阵渲染生成图像

实现步骤 #

• 首先确保你已经把三个R、G、B通道的矩阵转换为numpy二维数组，分别命名为r_mat、g_mat、b_mat，三者尺寸必须完全一致。如果原始数据是列表格式，先执行转换：r_mat = np.array(你的R通道列表)
• 导入所需依赖库，Colab环境默认预装，无需额外安装：

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

• 合并三个单通道矩阵为标准RGB三通道数组：

# 按最后一维拼接，得到形状为 (高度, 宽度, 3) 的RGB浮点数组
rgb_arr = np.stack([r_mat, g_mat, b_mat], axis=-1)

• 方案1：直接在Colab中渲染显示图片

plt.imshow(rgb_arr)
plt.axis('off') # 可选：关闭坐标轴显示
plt.show()

• 方案2：生成可编辑、可存储的PIL图片对象

# 将0~1浮点值转为0~255的8位整型，适配常规图片存储格式
rgb_img = (rgb_arr * 255).astype(np.uint8)
img = Image.fromarray(rgb_img)
# 在Colab中显示图片
display(img)
# 保存图片到Colab当前工作目录
img.save("output_rgb.png")

注意事项 #

• 若矩阵数值存在超出0~1范围的异常值，可在拼接后加裁剪操作避免显示异常：rgb_arr = np.clip(rgb_arr, 0.0, 1.0)
• 三个通道的矩阵宽高必须完全匹配，否则拼接步骤会报错

内容的提问来源于stack exchange，提问作者Rajdeep Das