如何用OpenCV在Python中实现PureThermal热成像摄像头流传输？

问题根源 #

PureThermal 1.2.2输出的是16位单通道原始热成像数据（像素值对应温度，范围通常为0_{65535），而OpenCV的`imshow`默认仅能正确渲染8位（0}255）图像。直接显示16位数据时，大部分像素值超出8位范围，导致窗口呈现黑色。

必要处理步骤 #

需要对读取的热成像帧做格式转换、归一化，可选添加伪彩色映射优化视觉效果，同时确保摄像头读取API适配MacOS环境：

• 指定MacOS摄像头读取API：MacOS下需显式使用CAP_AVFOUNDATION API读取外接摄像头，避免兼容性问题。
• 转换为16位单通道数组：确保读取的帧被解析为正确的16位格式。
• 归一化到8位范围：将0_{65535的温度值映射到0}255，适配OpenCV的显示要求。
• 伪彩色映射（可选）：将灰度热图转换为彩色热图，更直观区分温度差异。

修改后的完整代码 #

import numpy as np
import cv2

# 外接热成像摄像头索引
camera_index = 2
# MacOS下指定CAP_AVFOUNDATION API读取摄像头
cap = cv2.VideoCapture(camera_index, cv2.CAP_AVFOUNDATION)
# 设置帧率（PureThermal 1.2.2最大支持60FPS）
cap.set(cv2.CAP_PROP_FPS, 60)

if not cap.isOpened():
    print("无法打开热成像摄像头")
    exit()

while True:
    ret, frame = cap.read()
    if not ret:
        print("无法获取帧（流已结束？），退出...")
        break

    # 将帧转换为16位单通道数组（PureThermal输出的原始格式）
    # 若读取的是3通道帧，需提取单通道；直接读取为单通道则跳过此步
    if len(frame.shape) == 3:
        thermal_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY).astype(np.uint16)
    else:
        thermal_frame = frame.astype(np.uint16)

    # 归一化到8位范围（0~255）
    # 可选：根据实际温度范围调整min/max，比如用当前帧的最小/最大值自适应归一化
    normalized_frame = cv2.normalize(thermal_frame, None, 0, 255, cv2.NORM_MINMAX, dtype=cv2.CV_8U)

    # 可选伪彩色映射（推荐使用COLORMAP_JET，视觉效果清晰）
    colored_frame = cv2.applyColorMap(normalized_frame, cv2.COLORMAP_JET)

    # 显示处理后的热成像帧
    cv2.imshow("热成像画面", colored_frame)

    # 按q键退出
    if cv2.waitKey(1) == ord('q'):
        break

# 释放资源
cap.release()
cv2.destroyAllWindows()

关键代码说明 #

• cv2.VideoCapture(camera_index, cv2.CAP_AVFOUNDATION)：MacOS环境下必须指定此API，否则可能无法正确识别外接摄像头的16位输出格式。
• cv2.normalize(...)：将16位的温度数据线性映射到8位，确保所有像素值都能被imshow正确渲染。
• cv2.applyColorMap(...)：将灰度热图转换为彩色，不同颜色对应不同温度区间，比灰度图更易分辨温度差异。

内容的提问来源于stack exchange，提问作者Ashton T. Sperry