nicetoolbox.evaluation.metrics.point_cloud.Jpe

class nicetoolbox.evaluation.metrics.point_cloud.Jpe(**kwargs: Any)

Bases: Metric

Calculates the per joint position error for each frame.

Initializes the metric. Subclasses can use kwargs for specific setup.

Methods

compute	Compute the final metric from the stored state.
get_axis3	Get the joint names the metric is concerned with.
reset	Reset the metric's internal state.
update	Computes the per joint position error and stores it.

compute() → Dict[Tuple[str, str, str], List[BatchResult] | AggregatedResult]

Compute the final metric from the stored state.

get_axis3(chunk: ChunkWorkItem) → List[str]

Get the joint names the metric is concerned with.

reset() → None

Reset the metric’s internal state.

update(preds: torch.Tensor, gts: torch.Tensor, meta_chunk: ChunkWorkItem, meta_frames: List[FrameInfo]) → None

Computes the per joint position error and stores it.