open3d墙面提取代码

2023-12-01 19:30
文章标签 代码 提取 墙面 open3d

本文主要是介绍open3d墙面提取代码,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

门窗洞口提取需拍平墙,根据X、Y、Z轴的axsi()值设定,拍平后可以进行下一步处理,方便点云开发者下一步操作。

 

附上代码

import open3d as o3d
import numpy as np
import matplotlib.pyplot as plt
from alpha_shapes import Alpha_Shaper, plot_alpha_shapedef extract_walls_pcd(pcd, axis='z', threshold=2e+1000):# 提取墙面的点云points = np.asarray(pcd.points)if axis == 'x':axis_index = 0elif axis == 'y':axis_index = 1elif axis == 'z':axis_index = 2else:raise ValueError("Invalid axis. Must be 'x', 'y', or 'z'.")wall_points = points[abs(points[:, axis_index]) < threshold]wall_pcd = o3d.geometry.PointCloud()wall_pcd.points = o3d.utility.Vector3dVector(wall_points)return wall_pcddef plane_fit(pcd):# 使用RANSAC平面拟合plane_model, inliers = pcd.segment_plane(distance_threshold=0.5, ransac_n=5, num_iterations=1000)return plane_model, inliersdef plane_projection(pcd, plane_model, inliers):# 将点云投影到拟合的平面上plane_pcd = pcd.select_by_index(inliers)projected_points = np.asarray(plane_pcd.points)plane_params = plane_model[:-1]d = plane_model[-1]array_1 = projected_points.dot(plane_params[:, np.newaxis]) + darray_1_reshaped = np.tile(array_1, (1, 3))# projected_points -= projected_points.dot(plane_params[:, np.newaxis]) + dprojected_points -= array_1_reshapedplane_pcd.points = o3d.utility.Vector3dVector(projected_points)return plane_pcdif __name__ == "__main__":# 读取点云文件pcd = o3d.io.read_point_cloud(r"C:\Users\zhaojunzhe\Desktop\pyRANSAC-3D-master\pyRANSAC-3D-master\tests\dataset\20230713.pcd")# 提取墙面点云并拍平wall_pcd_x = extract_walls_pcd(pcd, axis='x')flat_wall_pcd_x = plane_projection(wall_pcd_x, *plane_fit(wall_pcd_x))wall_pcd_y = extract_walls_pcd(pcd, axis='y')flat_wall_pcd_y = plane_projection(wall_pcd_y, *plane_fit(wall_pcd_y))wall_pcd_z = extract_walls_pcd(pcd, axis='z')flat_wall_pcd_z = plane_projection(wall_pcd_z, *plane_fit(wall_pcd_z))# 保存平面点云为独立的文件o3d.io.write_point_cloud("flat_wall_pcd_x.pcd", flat_wall_pcd_x)o3d.io.write_point_cloud("flat_wall_pcd_y.pcd", flat_wall_pcd_y)o3d.io.write_point_cloud("flat_wall_pcd_z.pcd", flat_wall_pcd_z)# 创建一个可视化窗口vis = o3d.visualization.Visualizer()vis.create_window()# 将处理后的点云添加到可视化窗口中# vis.add_geometry(flat_wall_pcd_x)# vis.add_geometry(flat_wall_pcd_y)vis.add_geometry(flat_wall_pcd_z)# 显示点云vis.run()vis.destroy_window()

 

这篇关于open3d墙面提取代码的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/442429

相关文章

利用Python调试串口的示例代码

《利用Python调试串口的示例代码》在嵌入式开发、物联网设备调试过程中,串口通信是最基础的调试手段本文将带你用Python+ttkbootstrap打造一款高颜值、多功能的串口调试助手,需要的可以了... 目录概述:为什么需要专业的串口调试工具项目架构设计1.1 技术栈选型1.2 关键类说明1.3 线程模

Python Transformers库(NLP处理库)案例代码讲解

《PythonTransformers库(NLP处理库)案例代码讲解》本文介绍transformers库的全面讲解,包含基础知识、高级用法、案例代码及学习路径,内容经过组织,适合不同阶段的学习者,对... 目录一、基础知识1. Transformers 库简介2. 安装与环境配置3. 快速上手示例二、核心模

Java的栈与队列实现代码解析

《Java的栈与队列实现代码解析》栈是常见的线性数据结构,栈的特点是以先进后出的形式,后进先出,先进后出,分为栈底和栈顶,栈应用于内存的分配,表达式求值,存储临时的数据和方法的调用等,本文给大家介绍J... 目录栈的概念(Stack)栈的实现代码队列(Queue)模拟实现队列(双链表实现)循环队列(循环数组

使用Python从PPT文档中提取图片和图片信息(如坐标、宽度和高度等)

《使用Python从PPT文档中提取图片和图片信息(如坐标、宽度和高度等)》PPT是一种高效的信息展示工具,广泛应用于教育、商务和设计等多个领域,PPT文档中常常包含丰富的图片内容,这些图片不仅提升了... 目录一、引言二、环境与工具三、python 提取PPT背景图片3.1 提取幻灯片背景图片3.2 提取

Python实现word文档内容智能提取以及合成

《Python实现word文档内容智能提取以及合成》这篇文章主要为大家详细介绍了如何使用Python实现从10个左右的docx文档中抽取内容,再调整语言风格后生成新的文档,感兴趣的小伙伴可以了解一下... 目录核心思路技术路径实现步骤阶段一:准备工作阶段二:内容提取 (python 脚本)阶段三:语言风格调

使用Java将DOCX文档解析为Markdown文档的代码实现

《使用Java将DOCX文档解析为Markdown文档的代码实现》在现代文档处理中,Markdown(MD)因其简洁的语法和良好的可读性,逐渐成为开发者、技术写作者和内容创作者的首选格式,然而,许多文... 目录引言1. 工具和库介绍2. 安装依赖库3. 使用Apache POI解析DOCX文档4. 将解析

一文详解如何在Python中从字符串中提取部分内容

《一文详解如何在Python中从字符串中提取部分内容》:本文主要介绍如何在Python中从字符串中提取部分内容的相关资料,包括使用正则表达式、Pyparsing库、AST(抽象语法树)、字符串操作... 目录前言解决方案方法一:使用正则表达式方法二:使用 Pyparsing方法三:使用 AST方法四:使用字

C++使用printf语句实现进制转换的示例代码

《C++使用printf语句实现进制转换的示例代码》在C语言中,printf函数可以直接实现部分进制转换功能,通过格式说明符(formatspecifier)快速输出不同进制的数值,下面给大家分享C+... 目录一、printf 原生支持的进制转换1. 十进制、八进制、十六进制转换2. 显示进制前缀3. 指

使用Python实现全能手机虚拟键盘的示例代码

《使用Python实现全能手机虚拟键盘的示例代码》在数字化办公时代,你是否遇到过这样的场景:会议室投影电脑突然键盘失灵、躺在沙发上想远程控制书房电脑、或者需要给长辈远程协助操作?今天我要分享的Pyth... 目录一、项目概述:不止于键盘的远程控制方案1.1 创新价值1.2 技术栈全景二、需求实现步骤一、需求

Java中Date、LocalDate、LocalDateTime、LocalTime、时间戳之间的相互转换代码

《Java中Date、LocalDate、LocalDateTime、LocalTime、时间戳之间的相互转换代码》:本文主要介绍Java中日期时间转换的多种方法,包括将Date转换为LocalD... 目录一、Date转LocalDateTime二、Date转LocalDate三、LocalDateTim