使用OpenCV导出NumPy矩阵至TIFF文件遇错求解决方案

问题根源 #

• OpenCV的imwrite仅支持有限的TIFF数据类型，float16不在支持范围内，因此写入失败。
• 尽管float32可被imwrite写入，但imread默认不支持读取32位浮点样本的TIFF，导致读取报错。

修复方案与替代方法 #

方法1：使用tifffile库（推荐） #

tifffile是专门处理TIFF文件的库，完美支持float16、float32等各类浮点格式。

先安装库：

pip install tifffile

示例代码：

import numpy as np
import tifffile

# 创建float32矩阵
data = np.full((128, 128), -0.1, dtype=np.float32)
# 写入TIFF文件
tifffile.imwrite('temp.tiff', data)
print('tiff file written')

# 读取并验证
image = tifffile.imread('temp.tiff')
print(image.dtype)  # 输出float32，确认数据类型正确
print(image[0,0])   # 输出-0.1，验证内容无误

若需要使用float16，仅需修改dtype即可：

data = np.full((128, 128), -0.1, dtype=np.float16)

方法2：使用Pillow库 #

Pillow同样支持浮点TIFF的读写，需确保安装的版本兼容相关格式。

先安装库：

pip install pillow

示例代码：

import numpy as np
from PIL import Image

# 创建float32矩阵
data = np.full((128, 128), -0.1, dtype=np.float32)
# 转换为Image对象并保存
Image.fromarray(data).save('temp.tiff')
print('tiff file written')

# 读取并验证
image = np.array(Image.open('temp.tiff'))
print(image.dtype)  # 输出float32
print(image[0,0])   # 输出-0.1

OpenCV相关补充说明 #

如果坚持使用OpenCV：

• 无法直接读取float32格式的TIFF，但可通过检查文件是否存在、文件大小是否合理判断写入状态，或用上述第三方库读取验证写入结果。
• OpenCV不支持float16格式的TIFF写入，此路径不可行。

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