gnn_tracking.postprocessing.clusterscanner

Module Contents

Classes

ClusterScanner	Base class for cluster scanners. Use any of its subclasses.
CombinedClusterScanner	Combine multiple ClusterScanner objects.

class gnn_tracking.postprocessing.clusterscanner.ClusterScanner(*args, **kwargs)

Bases: pytorch_lightning.core.mixins.hparams_mixin.HyperparametersMixin, abc.ABC

Base class for cluster scanners. Use any of its subclasses.

abstract __call__(data: torch_geometric.data.Data, out: dict[str, torch.Tensor], i_batch: int) → None

reset() → None

get_foms() → dict[str, Any]

class gnn_tracking.postprocessing.clusterscanner.CombinedClusterScanner(scanners: list[ClusterScanner])

Bases: ClusterScanner

Combine multiple ClusterScanner objects.

__call__(*args, **kwargs)

reset() → None

get_foms() → dict[str, Any]