百度360必应搜狗淘宝本站头条
当前位置:网站首页 > 技术分类 > 正文

Python:如何创建和可视化点云(python怎么做可视化)

ztj100 2024-11-10 13:13 15 浏览 0 评论

在本教程中,我们将介绍点云,并了解如何用Python创建和可视化点云。

1、简介

点云应用无处不在:机器人、自动驾驶汽车、辅助系统、医疗保健等。点云是一种适合处理现实世界数据的3D表示,特别是在需要场景/对象的几何形状时,如对象的距离、形状和大小。

点云是一组点,代表现实世界中的场景或空间中的对象。它是几何对象和场景的离散表示。换句话说,点云PCD是n个点的集合,其中每个点Pi用其3D坐标表示:


注意,还可以添加一些其他特征来描述点云,如RGB颜色、法线等。例如,可以添加RGB颜色来提供颜色信息。

2、点云生成

点云通常使用3D扫描仪(激光扫描仪、飞行时间扫描仪和结构光扫描仪)或计算机辅助设计(CAD)模型生成。在本教程中,我们将首先创建随机点云并将其可视化。然后,我们将使用Open3D库从3D表面采样点,从3D模型生成它。最后,我们将看到如何从RGB-D数据创建它们。

让我们从导入Python库开始:

import numpy as np
import matplotlib.pyplot as plt
import open3d as o3d

2.1 随机点云

最简单的方法是随机创建一个点云。注意,我们通常不会创建要处理的随机点,除非为GAN(生成对抗网络)创建噪声。

通常,点云由(n×3)数组表示,其中n是点的数量。让我们用5个随机点创建一个点云:

number_points = 5
pcd = np.random.rand(number_points, 3)  # uniform distribution over [0, 1)
print(pcd)

我们可以直接打印这些点,但效率不高,特别是在大多数应用中,如果点的数量很大的话。更好的方法是将它们显示在3D空间中。让我们用Matplotlib库来可视化它:

# Create Figure:
fig, ax = plt.subplots(subplot_kw={"projection": "3d"})
ax.scatter3D(pcd[:, 0], pcd[:, 1], pcd[:, 2])
# label the axes
ax.set_xlabel("X")
ax.set_ylabel("Y")
ax.set_zlabel("Z")
ax.set_title("Random Point Cloud")
# display:
plt.show()

2.2 采样点云

直接处理3D模型需要时间。因此,从它们的三维表面采样点云是一个潜在的解决方案。让我们首先从Open3D数据集中导入兔子模型:

bunny = o3d.data.BunnyMesh()
mesh = o3d.io.read_triangle_mesh(bunny.path)

或者以如下方式导入:

mesh = o3d.io.read_triangle_mesh("data/bunny.ply")

接下来,显示 3D 模型以查看其外观。您可以移动鼠标从不同的视点进行查看。

# Visualize:
mesh.compute_vertex_normals() # compute normals for vertices or faces
o3d.visualization.draw_geometries([mesh])

要对点云进行采样,有几种方法。在此示例中,我们从导入的网格中均匀地采样 1000 个点并将其可视化:

# Sample 1000 points:
pcd = mesh.sample_points_uniformly(number_of_points=1000)

# visualize:
o3d.visualization.draw_geometries([pcd])

我们可以将创建的点云保存为 .ply 格式,如下所示:

# Save into ply file:
o3d.io.write_point_cloud("output/bunny_pcd.ply", pcd)

2.3 来自 RGB-D 数据的点云

RGB-D 数据是使用RGB-D传感器(例如 Microsoft Kinect)收集的,该传感器同时提供 RGB 图像和深度图像。RGB-D传感器被广泛应用于室内导航、避障等领域。由于RGB图像提供像素颜色,所以深度图像的每个像素表示其与相机的距离。

Open3D 为 RGB-D 图像处理提供了一组函数。要使用 Open3D 函数从 RGB-D 数据创建点云,只需导入两个图像,创建一个 RGB-D 图像对象,最后计算点云如下:

# read the color and the depth image:
color_raw = o3d.io.read_image("../data/rgb.jpg")
depth_raw = o3d.io.read_image("../data/depth.png")

# create an rgbd image object:
rgbd_image = o3d.geometry.RGBDImage.create_from_color_and_depth(
    color_raw, depth_raw, convert_rgb_to_intensity=False)
# use the rgbd image to create point cloud:
pcd = o3d.geometry.PointCloud.create_from_rgbd_image(
    rgbd_image,
    o3d.camera.PinholeCameraIntrinsic(
        o3d.camera.PinholeCameraIntrinsicParameters.PrimeSenseDefault))

# visualize:
o3d.visualization.draw_geometries([pcd])

3、Open3D和NumPy

有时您需要在Open3D和NumPy表示之间切换。例如,假设我们想要将NumPy点云转换为Open3D.PointCloud对象进行可视化,并使用Matplotlib可视化兔子的3D模型。

3.1 从NumPy到Open3D

在本例中,我们使用NumPy.random.rand()函数创建2000个随机点,该函数从[0,1]的均匀分布中创建随机样本。然后我们创建一个Open3D.PointCloud对象,并使用Open3D.utility.Vector3dVector()函数将其Open3D.PointCloud.points特征设置为随机点。

# Create numpy pointcloud:
number_points = 2000
pcd_np = np.random.rand(number_points, 3)

# Convert to Open3D.PointCLoud:
pcd_o3d = o3d.geometry.PointCloud()  # create point cloud object
pcd_o3d.points = o3d.utility.Vector3dVector(pcd_np)  # set pcd_np as the point cloud points

# Visualize:
o3d.visualization.draw_geometries([pcd_o3d])

3.2 从 Open3D到NumPy

这里,我们首先使用Open3D.io.read_point_cloud()函数从.ply文件中读取点云,该函数返回一个Open3D.PointCloud对象。现在我们只需要使用NumPy.asarray()函数将表示点的Open3D.PointCloud.points特征转换为NumPy数组。最后,我们像上面那样显示获得的数组。

# Read the bunny point cloud file:
pcd_o3d = o3d.io.read_point_cloud("../data/bunny_pcd.ply")

# Convert the open3d object to numpy:
pcd_np = np.asarray(pcd_o3d.points)

# Display using matplotlib:
fig, ax = plt.subplots(subplot_kw={"projection": "3d"})
ax.scatter3D(pcd_np[:, 0], pcd_np[:, 2], pcd_np[:, 1])
# label the axes
ax.set_xlabel("X")
ax.set_ylabel("Y")
ax.set_zlabel("Z")
ax.set_title("Bunny Point Cloud")
# display:
plt.show()

4、最后

在本教程中,我们学习了如何创建和可视化点云。在接下来的教程中,我们将学习如何处理它们。

相关推荐

再说圆的面积-蒙特卡洛(蒙特卡洛方法求圆周率的matlab程序)

在微积分-圆的面积和周长(1)介绍微积分方法求解圆的面积,本文使用蒙特卡洛方法求解圆面积。...

python编程:如何使用python代码绘制出哪些常见的机器学习图像?

专栏推荐...

python创建分类器小结(pytorch分类数据集创建)

简介:分类是指利用数据的特性将其分成若干类型的过程。监督学习分类器就是用带标记的训练数据建立一个模型,然后对未知数据进行分类。...

matplotlib——绘制散点图(matplotlib散点图颜色和图例)

绘制散点图不同条件(维度)之间的内在关联关系观察数据的离散聚合程度...

python实现实时绘制数据(python如何绘制)

方法一importmatplotlib.pyplotaspltimportnumpyasnpimporttimefrommathimport*plt.ion()#...

简单学Python——matplotlib库3——绘制散点图

前面我们学习了用matplotlib绘制折线图,今天我们学习绘制散点图。其实简单的散点图与折线图的语法基本相同,只是作图函数由plot()变成了scatter()。下面就绘制一个散点图:import...

数据分析-相关性分析可视化(相关性分析数据处理)

前面介绍了相关性分析的原理、流程和常用的皮尔逊相关系数和斯皮尔曼相关系数,具体可以参考...

免费Python机器学习课程一:线性回归算法

学习线性回归的概念并从头开始在python中开发完整的线性回归算法最基本的机器学习算法必须是具有单个变量的线性回归算法。如今,可用的高级机器学习算法,库和技术如此之多,以至于线性回归似乎并不重要。但是...

用Python进行机器学习(2)之逻辑回归

前面介绍了线性回归,本次介绍的是逻辑回归。逻辑回归虽然名字里面带有“回归”两个字,但是它是一种分类算法,通常用于解决二分类问题,比如某个邮件是否是广告邮件,比如某个评价是否为正向的评价。逻辑回归也可以...

【Python机器学习系列】拟合和回归傻傻分不清?一文带你彻底搞懂

一、拟合和回归的区别拟合...

推荐2个十分好用的pandas数据探索分析神器

作者:俊欣来源:关于数据分析与可视化...

向量数据库:解锁大模型记忆的关键!选型指南+实战案例全解析

本文较长,建议点赞收藏,以免遗失。更多AI大模型应用开发学习视频及资料,尽在...

用Python进行机器学习(11)-主成分分析PCA

我们在机器学习中有时候需要处理很多个参数,但是这些参数有时候彼此之间是有着各种关系的,这个时候我们就会想:是否可以找到一种方式来降低参数的个数呢?这就是今天我们要介绍的主成分分析,英文是Princip...

神经网络基础深度解析:从感知机到反向传播

本文较长,建议点赞收藏,以免遗失。更多AI大模型应用开发学习视频及资料,尽在...

Python实现基于机器学习的RFM模型

CDA数据分析师出品作者:CDALevelⅠ持证人岗位:数据分析师行业:大数据...

取消回复欢迎 发表评论: