函数extract_dir用于从TUM RGB-D数据集中提取指定时间点的深度图和彩色图像。
open3d.data.SampleTUMRGBDImage.extract_dir(dataset_path, time_str)
dataset_path
:字符串类型,指定数据集的路径。time_str
:字符串类型,指定时间点的序列编号,格式为yyyy-mm-dd-HH-MM-SS。如果指定时间点的深度图和彩色图像存在,函数会将它们分别存储为Open3D的geometry.Image类型和geometry.RGBDImage类型,并以元组的形式进行返回。
import open3d as o3d
# 设置数据集路径
dataset_path = "/path/to/dataset/"
# 提取时间点为2021-11-01-10-30-00的数据
time_str = "2021-11-01-10-30-00"
depth, color = o3d.data.SampleTUMRGBDImage.extract_dir(dataset_path, time_str)
# 显示深度图
o3d.visualization.draw_geometries([depth])
# 显示彩色图像
o3d.visualization.draw_geometries([color])