1 vin.encoders.pointnext_utils

vin.encoders.pointnext_utils

Utility helpers for the optional PointNeXt VIN encoder.

aria_nbv.vin.encoders.pointnext.PointNeXtSEncoder wraps an OpenPoints PointNeXt-S model for semidense point-cloud context. This sidecar owns the unstructured OpenPoints/checkpoint glue so the torch.nn.Module implementation can stay focused on module lifecycle and tensor contracts.

1.1 Functions

Name Description
extract_point_encoder_tensor Return the first tensor-like feature payload from an OpenPoints output.
load_checkpoint_strict Load an OpenPoints-style checkpoint with strict PyTorch key matching.
load_pointnext_cfg Load a PointNeXt/OpenPoints YAML config with openpoints.utils.EasyConfig.

1.1.1 extract_point_encoder_tensor

vin.encoders.pointnext_utils.extract_point_encoder_tensor(output)

Return the first tensor-like feature payload from an OpenPoints output.

1.1.1.1 Parameters

Name Type Description Default
output Any Tensor, mapping, tuple, or list returned by an external point encoder. OpenPoints variants commonly use keys such as feat or features but checkpoint/model wrappers are not fully uniform. required

1.1.1.2 Returns

Name Type Description
Tensor The first torch.Tensor feature payload found.

1.1.1.3 Raises

Name Type Description
TypeError If no tensor payload can be located.

1.1.2 load_checkpoint_strict

vin.encoders.pointnext_utils.load_checkpoint_strict(model, checkpoint_path)

Load an OpenPoints-style checkpoint with strict PyTorch key matching.

1.1.2.1 Parameters

Name Type Description Default
model nn.Module PointNeXt module built by OpenPoints. required
checkpoint_path Path Path to a checkpoint containing a state dict directly or under common wrapper keys such as model or state_dict. required

1.1.2.2 Raises

Name Type Description
RuntimeError If the checkpoint does not contain a dictionary state payload after unwrapping common OpenPoints keys.

1.1.3 load_pointnext_cfg

vin.encoders.pointnext_utils.load_pointnext_cfg(cfg_path)

Load a PointNeXt/OpenPoints YAML config with openpoints.utils.EasyConfig.

1.1.3.1 Parameters

Name Type Description Default
cfg_path Path Absolute path to the OpenPoints YAML configuration file. required

1.1.3.2 Returns

Name Type Description
Any Parsed OpenPoints config object.
Back to top