本文共 4184 字，大约阅读时间需要 13 分钟。

降采样作业：

# 实现voxel滤波，并加载数据集中的文件进行验证import open3d as o3d import osimport numpy as npfrom pyntcloud import PyntCloudimport matplotlib.pyplot as pltimport randomfrom pandas import DataFrame# matplotlib显示点云函数def Point_Cloud_Show(points):    fig = plt.figure(dpi=150)    ax = fig.add_subplot(111, projection='3d')    ax.scatter(points[:, 0], points[:, 1], points[:, 2], cmap='spectral', s=2, linewidths=0, alpha=1, marker=".")    plt.title('Point Cloud')    ax.set_xlabel('x')    ax.set_ylabel('y')    ax.set_zlabel('z')    plt.show()# 二维点云显示函数def Point_Show(pca_point_cloud):    x = []    y = []    pca_point_cloud = np.asarray(pca_point_cloud)    for i in range(10000):        x.append(pca_point_cloud[i][0])        y.append(pca_point_cloud[i][1])    plt.scatter(x, y)    plt.show()# 功能：对点云进行voxel滤波# 输入：#     point_cloud：输入点云#     leaf_size: voxel尺寸def voxel_filter(point_cloud, leaf_size,filter_mode):    filtered_points = []    # 作业3    # 屏蔽开始    #step1 计算边界点    x_max, y_max, z_max = np.amax(point_cloud,axis=0)      #计算 x,y,z三个维度的最值    x_min, y_min, z_min = np.amin(point_cloud, axis=0)    #step2 确定体素的尺寸    size_r = leaf_size    #step3 计算每个 volex的维度    Dx = (x_max - x_min)/size_r    Dy = (y_max - y_min)/size_r    Dz = (z_max - z_min)/size_r    #step4 计算每个点在volex grid内每一个维度的值    h = list()    for i in range(len(point_cloud)):        hx = np.floor((point_cloud[i][0] - x_min)/size_r)        hy = np.floor((point_cloud[i][1] - y_min)/size_r)        hz = np.floor((point_cloud[i][2] - z_min)/size_r)        h.append(hx + hy*Dx + hz*Dx*Dy)    #step5 对h值进行排序    h = np.array(h)    h_indice  = np.argsort(h)   #提取索引    h_sorted = h[h_indice]      #升序    count = 0 #用于维度的累计        np.seterr(divide='ignore',invalid='ignore') #忽略除法遇到无效值的问题    #将h值相同的点放入到同一个grid中，并进行筛选    for i  in range(len(h_sorted)-1):      #数据点        if h_sorted[i] == h_sorted[i+1]:   #当前的点与后面的相同，放在同一个volex grid中            continue        else:                filter_mode == "random"  #随机滤波                point_idx = h_indice[count: i+1]                random_points =  random.choice(point_cloud[point_idx])                filtered_points.append(random_points)                count = i    for i  in range(len(h_sorted)-1):      #数据点        if h_sorted[i] == h_sorted[i+1]:   #当前的点与后面的相同，放在同一个volex grid中            continue        else:                        filter_mode == "centroid"    #均值滤波            point_idx = h_indice[count: i+1]            filtered_points.append(np.mean(point_cloud[point_idx],axis=0))   #取同一个grid的均值            count = i    # 屏蔽结束    # 把点云格式改成array，并对外返回    filtered_points = np.array(filtered_points, dtype=np.float64)    return filtered_pointsdef main():    # # 从ModelNet数据集文件夹中自动索引路径，加载点云    # cat_index = 10 # 物体编号，范围是0-39，即对应数据集中40个物体    # root_dir = '/Users/renqian/cloud_lesson/ModelNet40/ply_data_points' # 数据集路径    # cat = os.listdir(root_dir)    # filename = os.path.join(root_dir, cat[cat_index],'train', cat[cat_index]+'_0001.ply') # 默认使用第一个点云    # point_cloud_pynt = PyntCloud.from_file(file_name)    # 加载自己的点云文件    point_cloud_raw = np.genfromtxt(r"car_0001.txt", delimiter=",")    point_cloud_raw = DataFrame(point_cloud_raw[:,0:3])   # 为 xyz的 N*3矩阵    point_cloud_raw.columns = ['x', 'y', 'z']  # 给选取到的数据 附上标题    point_cloud_pynt = PyntCloud(point_cloud_raw)  # 将points的数据 存到结构体中    point_cloud_o3d_orign = point_cloud_pynt.to_instance("open3d", mesh=False)  # to_instance实例化    point_cloud_o3d_filter_1 = o3d.geometry.PointCloud()     #实例化        point_cloud_o3d_filter_2 = o3d.geometry.PointCloud()     #实例化         points = np.array(point_cloud_o3d_orign.points)    # 调用voxel滤波函数，实现滤波    filtered_cloud_1 = voxel_filter(points, 0.05, "random")   #random    point_cloud_o3d_filter_1.points = o3d.utility.Vector3dVector(filtered_cloud_1)    filtered_cloud_2 = voxel_filter(points, 0.05, "centroid")   #centroid     point_cloud_o3d_filter_2.points = o3d.utility.Vector3dVector(filtered_cloud_2)    # 显示滤波前后的点云    o3d.visualization.draw_geometries([point_cloud_o3d_orign])# 显示原始点云    o3d.visualization.draw_geometries([point_cloud_o3d_filter_1])    o3d.visualization.draw_geometries([point_cloud_o3d_filter_2])if __name__ == '__main__':    main()

原图：

转载地址：http://vcexi.baihongyu.com/