Skip to content

simple

Single forward-pass inference over the whole scene.

Classes:

  • SimpleInferer –

    Direct call to the predictor on the whole scene.

SimpleInferer 

SimpleInferer(softmax: bool = False)

Bases: Inferer

Direct call to the predictor on the whole scene.

The lightest possible Inferer. Use when the model can consume the entire point cloud in one forward pass (object classification, small scenes).

Parameters:

  • softmax (bool, default: False ) –

    If True, softmax the predictor output over the last dim before returning it. The default returns the predictor output unchanged.

Example 
from torch_pointcloud.inferers import SimpleInferer

inferer = SimpleInferer()
logits = inferer(data, predictor=lambda d: model(d["pos"], d["pos"], d["batch"]))