当前位置: 首页 > 工具软件 > ONI > 使用案例 >

怎么样用Python 读取oni 格式文件?

罗学真
2023-12-01

什么是oni 文件?

oni 文件是openni 的存储文件。

Python读取方法

先安装pyopenni 库

pip install openni

然后去下载OpenNI 2 SDK,链接: OpenNI2 SDK.
下载好后把sdk放进代码同目录下。
新建一个test.py,代码如下

from openni import openni2
openni2.initialize()     # can also accept the path of the OpenNI redistribution

如果能运行成功,则OpenNI的环境搭建好了。
如果不能成功,提示缺少文件,则把sdk中的文件放入对应位置。

现在可以用OpenNI直接读入oni文件。下面是示例

python 代码示例

from openni import openni2
import cv2
import numpy as np

openni2.initialize()
dev = openni2.Device.open_file('20190102.oni'.encode('utf8'))
# 创建读入流
depth_stream = dev.create_depth_stream()
color_stream = dev.create_color_stream()

# 这里深度图和彩色图的图片总数可能不一样
print(depth_stream.get_number_of_frames())
print(color_stream.get_number_of_frames())

depth_stream.start()
color_stream.start()

# 循环所有图片一次,否则会一直循环。
for i in range(depth_stream.get_number_of_frames()):
    # f_d 是一个一维的流,shape是【307200】 ,就是 480* 640
    f_d = depth_stream.read_frame().get_buffer_as_uint16()
    # c_f 是一个一维的流,shape是【921600】 ,就是 480* 640 *3
    c_f = color_stream.read_frame().get_buffer_as_uint8()

    # 先处理深度流
    dep = np.frombuffer(f_d, dtype=np.uint16)
    # [307200] => [480,640]
    dep = np.reshape(dep, (480, 640))
    # 转成 0~1的浮点数
    dep = dep.astype(np.float64) / np.max(dep)
    cv2.imshow("depth", dep)

    # 处理彩色流
    img = np.frombuffer(c_f, dtype=np.uint8)
    # [921600] => [480,640,3]
    img = np.reshape(img, (480, 640, 3))
    # RGB =>  BGR 
    cv2.imshow("color", cv2.cvtColor(img, cv2.COLOR_RGB2BGR))

    cv2.waitKey(20)


depth_stream.close()
color_stream.close()

cv2.destroyAllWindows()

openni2.unload()
 类似资料: