python大恒相机保存RAW图和实时显示

fmdpenny2025-02-14 7:08
一、用python进行采集
临时有个任务需要进行数据采集
二、直接上代码
python 复制代码
import gxipy as gx
import cv2
import numpy as np

def main():
    # 初始化设备管理器
    device_manager = gx.DeviceManager()

    # 枚举设备
    dev_num, dev_info_list = device_manager.update_device_list()
    print(dev_info_list)
    if dev_num == 0:
        print("没有找到设备")
        return

    # 获取第一个设备的序列号并打开设备
    str_sn = dev_info_list[0].get("sn")
    cam = device_manager.open_device_by_sn(str_sn)

    # # 设置相机参数
    # cam.ExposureTime.set(500)  # 设置曝光时间
    # cam.Gain.set(1.0)  # 设置增益
    exposure_time = 230000  # 例如，设置为10000微秒（10毫秒）
    cam.ExposureTime.set(exposure_time)
    # 获取当前曝光时间以确认设置成功
    current_exposure_time = cam.ExposureTime.get()
    print(f"当前曝光时间: {current_exposure_time} 微秒")

    # 开始采集
    cam.stream_on()
    # 2592*2048
    try:
        while True:

            # 设置曝光时间（单位：微秒）

            # 从第 0 个流通道获取一幅图像
            raw_image = cam.data_stream[0].get_image()

            if raw_image is None:
                continue
            # 将RAW图像数据转换为numpy数组
            raw_image_np = raw_image.get_numpy_array()
            # 从彩色原始图像获取 RGB 图像
            rgb_image = raw_image.convert("RGB")

            if rgb_image is None:
                continue

            # 从 RGB 图像数据创建 numpy 数组
            numpy_image = rgb_image.get_numpy_array()

            if numpy_image is None:
                continue

            # 将图像从 RGB 转换为 BGR，因为 OpenCV 使用 BGR 格式
            bgr_image = cv2.cvtColor(numpy_image, cv2.COLOR_RGB2BGR)
            # 显示提取的区域
            cv2.namedWindow('Real-Time Image', cv2.WINDOW_NORMAL)
            # 显示图像
            cv2.imshow("Real-Time Image", bgr_image)

            # 按下 'q' 键退出循环
            if cv2.waitKey(1) & 0xFF == ord('q'):
                break
            elif cv2.waitKey(1) & 0xFF == ord('s'):
                cv2.imwrite('saved_image.png', bgr_image)
                # 保存RAW图像
                raw_image_np.tofile("raw_image_2856_470_23_G15.raw")
                print("RAW图像已保存为 raw_image.raw")

    finally:
        # 停止采集并关闭设备
        cam.stream_off()
        cam.close_device()

        # 关闭所有 OpenCV 窗口
        cv2.destroyAllWindows()

if __name__ == "__main__":
    main()