(coord, color=None, file_path="pc.ply", logger=None)
| 153 | |
| 154 | |
| 155 | def save_point_cloud(coord, color=None, file_path="pc.ply", logger=None): |
| 156 | os.makedirs(os.path.dirname(file_path), exist_ok=True) |
| 157 | coord = to_numpy(coord) |
| 158 | if color is not None: |
| 159 | color = to_numpy(color) |
| 160 | pcd = o3d.geometry.PointCloud() |
| 161 | pcd.points = o3d.utility.Vector3dVector(coord) |
| 162 | pcd.colors = o3d.utility.Vector3dVector(np.ones_like(coord) if color is None else color) |
| 163 | o3d.io.write_point_cloud(file_path, pcd) |
| 164 | if logger is not None: |
| 165 | logger.info(f"Save Point Cloud to: {file_path}") |
| 166 | |
| 167 | |
| 168 | def remove_small_group(group_ids, th): |
no test coverage detected