Method get_mesh

pointersect/inference/structures.py:360–417

Reconstruct a mesh from the point cloud. Args: bidx: the batch index. method: 'poisson': use Poisson surface reconstruction 'alpha': use Alpha shapes 'ball': use Ball pivoting recomp

(
            self,
            bidx: int,
            method: str,
            recompute_normal: bool,
            alpha: float = 0.03,
            ball_radii: T.List[float] = (0.005, 0.01, 0.02, 0.04),
            poisson_depth: int = 8,
    )

        return o3d_pcds

    def get_mesh(
            self,
            bidx: int,
            method: str,
            recompute_normal: bool,
            alpha: float = 0.03,
            ball_radii: T.List[float] = (0.005, 0.01, 0.02, 0.04),
            poisson_depth: int = 8,
    ) -> 'Mesh':
        """
        Reconstruct a mesh from the point cloud.

        Args:
            bidx:
                the batch index.
            method:
                'poisson': use Poisson surface reconstruction
                'alpha': use Alpha shapes
                'ball': use Ball pivoting
            recompute_normal:
                whether to recompute the vertex normal (assuming no normal at the points)
            alpha:
                alpha value used by alpha shapes
            ball_radii:
                radii used by ball pivot.
                radii of the individual balls that are pivoted on the point cloud.
                When the ball touches three points, a triangle is created.
            poisson_depth:
                depth used by poisson reconstruction.
                A higher depth value means a mesh with more details.
        """

        o3d_pcd: o3d.geometry.PointCloud = self.get_o3d_pcds(
            estimate_normal_if_not_exist=True,
        )[bidx]

        if recompute_normal:
            o3d_pcd.estimate_normals()

        if method == 'alpha':
            o3d_mesh = o3d.geometry.TriangleMesh.create_from_point_cloud_alpha_shape(
                o3d_pcd, alpha)
        elif method == 'ball':
            o3d_mesh = o3d.geometry.TriangleMesh.create_from_point_cloud_ball_pivoting(
                o3d_pcd, o3d.utility.DoubleVector(ball_radii),
            )
        elif method == 'poisson':
            o3d_mesh, densities = o3d.geometry.TriangleMesh.create_from_point_cloud_poisson(
                o3d_pcd, depth=poisson_depth)
        else:
            raise NotImplementedError

        return Mesh(
            mesh=o3d_mesh,
            scale=None,
            center_w=None,
            preprocess_mesh=False,
        )