Python中将结构化RGB dtype数组转换为float32数组及报错解决方案

你自定义的RGB_DTYPE属于numpy结构化数据类型，每个像素点是包含R、G、B三个独立uint8字段的复合结构，而非常规的三维数组布局。numpy不支持直接将复合结构化类型批量转换为单精度浮点的基础类型，默认类型转换规则判定该操作属于不安全转换，因此抛出对应错误。

你需要先将结构化数组转换为常规的三通道uint8数组，再执行类型转换，以下两种方案均可实现：

• 方案1：字段提取堆叠（兼容性强，不受内存布局影响）

import numpy as np

x_rgb = load_rgb_image(out_fname)
# 依次提取R/G/B三个字段的数值，沿最后一个维度堆叠后转float32
x = np.stack([x_rgb['R'], x_rgb['G'], x_rgb['B']], axis=-1).astype(np.float32)

# 如果需要将数值归一化到0~1区间，补充以下操作即可
# x /= 255.0

• 方案2：内存视图转换（无数据拷贝，执行效率更高，要求字段内存连续顺序为R/G/B）

import numpy as np

x_rgb = load_rgb_image(out_fname)
# 直接将结构化数组的内存重新解释为uint8类型，自动得到三通道数组后转float32
x = x_rgb.view(np.uint8).astype(np.float32)

# 同样可按需补充归一化操作
# x /= 255.0

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