pts_2d_ori = contour_info["pts_2d"] pts_3d_ori = pm[pts_2d_ori[:, 0], pts_2d_ori[:, 1], :] pts_3d = pts_3d_ori[np.where(~np.isnan(pts_3d_ori[:, 0]))] pts_2d = pts_2d_ori[np.where(~np.isnan(pts_3d_ori[:, 0]))] pts_2d_ori = contour_info["pts_2d"] pts_3d_ori = pm[pts_2d_ori[:, 0], pts_2d_ori[:, 1], :] pts_3d = pts_3d_ori[np.where(~np.isnan(pts_3d_ori[:, 0]))] pts_2d = pts_2d_ori[np.where(~np.isnan(pts_3d_ori[:, 0]))]
时间: 2024-03-27 15:41:57 浏览: 61
一个2D转3D的方法
这段代码的功能是将一个三维点云中的 NaN 值去除,并将点云的像素坐标和三维坐标分别保存到两个数组 pts_2d 和 pts_3d 中。
具体来说,首先从 contour_info 中取出轮廓的像素坐标 pts_2d_ori。然后通过 pm 数组和 pts_2d_ori 计算出每个像素点对应的三维坐标 pts_3d_ori。接着,使用 np.where 函数找到 pts_3d_ori 中不包含 NaN 值的索引位置,并将这些位置对应的三维坐标保存到 pts_3d 数组中。同时,将这些位置对应的像素坐标保存到 pts_2d 数组中。
最后,pts_2d 和 pts_3d 分别保存了点云的像素坐标和三维坐标,但是这些点云的数量可能会比原始的轮廓点数少,因为去除了 NaN 值。
阅读全文