Method predict_sdf

point_e/models/sdf.py:61–69

Predict the SDF at the query points given the encoded point clouds. Each query point should be treated independently, only conditioning on the point clouds themselves.

(
        self, x: torch.Tensor, encoded: Optional[Dict[str, torch.Tensor]]
    )

    @abstractmethod
    def predict_sdf(
        self, x: torch.Tensor, encoded: Optional[Dict[str, torch.Tensor]]
    ) -> torch.Tensor:
        """
        Predict the SDF at the query points given the encoded point clouds.

        Each query point should be treated independently, only conditioning on
        the point clouds themselves.
        """


class CrossAttentionPointCloudSDFModel(PointCloudSDFModel):