如何在DigitalMicrograph中用Python保存RGB彩色图像为PNG文件

原脚本仅支持单通道DM图像保存为PNG，遇到RGB图像时因GetNumArray()不支持而失败，可通过以下方案解决：

核心思路 #

1. 先判断当前DM图像的通道数，区分单通道和RGB三通道图像
2. 对RGB图像，分别提取红、绿、蓝三个通道的单通道数据
3. 将三个通道数据合并为标准RGB格式数组后保存

修改后的完整代码 #

import tkinter as tk
from tkinter import filedialog
from skimage import io, img_as_ubyte
import numpy as np
import DigitalMicrograph as DM

# 隐藏tkinter主窗口
root = tk.Tk()
root.withdraw()

# 获取当前激活的DM图像
img = DM.GetFrontImage()
num_channels = img.GetNumChannels()

# 根据通道数处理图像数据
if num_channels == 3:
    # 提取RGB三个通道的单通道图像数据
    r_img = img.GetImageForChannel(0)
    g_img = img.GetImageForChannel(1)
    b_img = img.GetImageForChannel(2)
    
    r_data = r_img.GetNumArray()
    g_data = g_img.GetNumArray()
    b_data = b_img.GetNumArray()
    
    # 合并为RGB格式数组（形状为[高度, 宽度, 3]）
    img_data = np.stack([r_data, g_data, b_data], axis=-1)
else:
    # 沿用原粮图像Run表 �--HO           商海的单通道处理逻辑
    img_data = img.GetNumArray()

# 选择保存路径
save_path = filedialog.asksaveasfilename(
    title='保存PNG图像',
    defaultextension='.png',
    filetypes=[('PNG图像', '*.png')],
)

if save_path:
    try:
        # 转换为uint8格式（PNG支持的标准格式）
        img_u8 = img_as_ubyte(img_data)
        io.imsave(save_path, img_u8)
        print(f'图像已保存至: {save_path}')
    except OSError as e:
        print(f'保存失败: {e}')
else:
    print('用户取消保存操作')

关键代码说明 #

• img.GetNumChannels()：获取当前DM图像的通道数，以此判断是否为RGB图像
• img.GetImageForChannel(index)：按索引提取对应通道的单通道图像，索引0对应红色、1对应绿色、2对应蓝色
• np.stack([...], axis=-1)：将三个单通道的二维数组合并为三维RGB数组，符合PNG图像的格式要求

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