import open3d as o3d import numpy as np pcd = o3d.io.read_point_cloud("data.pcd") points = np.asarray(pcd.points) bbox = pcd.get_axis_aligned_bounding_box() bbox_points = np.asarray(bbox.get_box_points()) vis = o3d.visualization.Visualizer() vis.create_window() vis.add_geometry(pcd) bbox_lines = [[0,1],[1,2],[2,3],[3,0], [4,5],[5,6],[6,7],[7,4], [0,4],[1,5],[2,6],[3,7]] bbox_colors = [[1,0,0] for i in range(len(bbox_lines))] line_set = o3d.geometry.LineSet( points=o3d.utility.Vector3dVector(bbox_points), lines=o3d.utility.Vector2iVector(bbox_lines), ) line_set.colors = o3d.utility.Vector3dVector(bbox_colors) vis.add_geometry(line_set) labels = np.zeros(points.shape[0]) eps = 0.1 min_points = 10 for i in range(points.shape[0]): if labels[i] != 0: continue neighbors = np.where(np.sum((points - points[i])2, axis=1) < eps2)[0] if neighbors.shape[0] < min_points: labels[i] = -1 else: labels[neighbors] = i+1 cluster_colors = [[np.random.uniform(0, 1), np.random.uniform(0, 1), np.random.uniform(0, 1)] for i in range(np.max(labels))] for i in range(np.max(labels)): if i == -1: continue cluster_points = points[labels==i,:] cluster_pcd = o3d.geometry.PointCloud() cluster_pcd.points = o3d.utility.Vector3dVector(cluster_points) cluster_pcd.paint_uniform_color(cluster_colors[i]) vis.add_geometry(cluster_pcd) vis.run() vis.destroy_window() open3d未响应