奥比中光相机的python采集代码

---

前言

---

在Windows系统下，使用python语言，采集奥比中光相机的拍摄的图像。

一、驱动安装

在奥比中光官网安装驱动。
下载地址：https://developer.orbbec.com.cn/download.html?id=32
进入这个页面点击“安装”------“下一页”------“完成"。

进入设备管理器查看，奥比中光相机正常运转。

二、配置openni

使用python驱动奥比中光相机需要借助openni2，安装步骤如下。
从官网下载openni2安装包，地址：https://structure.io/openni

安装完成后，系统会自动配置环境变量。

如果环境变量不存在，需要自己手动补齐。
在官网下载的openni2不包含orbbec.dll，orbbec.ini。手动将这两个文件复制到刚才安装的OpenNI2\Redist\OpenNI2\Drivers文件夹中。

三、采集代码

from openni import openni2
import numpy as np
import cv2
import datetime
import matplotlib.pyplot as plt
import time


def loop_func(func, second):
    while True:
        func()
        time.sleep(second)
        key = cv2.waitKey(1)
        if int(key) == ord('Q'):
            break


def picture_capture():
    frame = depth_stream.read_frame()
    dframe_data = np.array(frame.get_buffer_as_triplet()).reshape([480, 640, 2])
    dpt1 = np.asarray(dframe_data[:, :, 0], dtype='float32')
    dpt2 = np.asarray(dframe_data[:, :, 1], dtype='float32')

    dpt2 *= 255
    dpt = dpt1 + dpt2

    now = datetime.datetime.now()
    now = now.strftime('%Y%m%d%H%M%S')
    list1 = ['E:/testpic/', now, 'dth', '.png']
    list2 = ['E:/testpic/', now, 'rgb', '.png']
    address1 = ''.join(list1)
    address2 = ''.join(list2)
    ret, frame = cap.read()
    cv2.imwrite(address2, frame)
    print(dpt.shape)
    plt.figure(640)
    plt.imshow(dpt)
    plt.axis('off')  
    fig = plt.gcf()
    fig.set_size_inches(6.4 / 3, 4.81 / 3)  # dpi = 300, output = 700*700 pixels
    plt.gca().xaxis.set_major_locator(plt.NullLocator())
    plt.gca().yaxis.set_major_locator(plt.NullLocator())
    plt.subplots_adjust(top = 1, bottom = 0, right = 1, left = 0, hspace = 0, wspace = 0)
    plt.margins(0,0)
    fig.savefig(address1, format='png', transparent=True, dpi=300, pad_inches = 0)


if __name__ == "__main__":

    openni2.initialize()

    dev = openni2.Device.open_any()
    print(dev.get_device_info())

    depth_stream = dev.create_depth_stream()
    depth_stream.start()

    cap = cv2.VideoCapture(0)

    loop_func(picture_capture, 1)
    depth_stream.stop()
    dev.close()

使用以上代码就可以实时采集图片。