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.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]#