mlreco.post_processing.metrics.cluster_cnn_metrics module

mlreco.post_processing.metrics.cluster_cnn_metrics.cluster_cnn_metrics(cfg, module_cfg, data_blob, res, logdir, iteration, data_idx=None, seg_label=None, clust_data=None, particles=None, embeddings=None, margins=None, seediness=None, ghost_mask=None, true_ghost_mask=None, seg_label_noghost=None, clust_data_noghost=None, seg_prediction=None, **kwargs)[source]

Compute metrics for SPICE stage (CNN particle instance clustering).

TODO assumes ghost points for now

Parameters

  • data_blob (dict) – The input data dictionary from iotools.

  • res (dict) – The output of the network, formatted using analysis_keys.

  • cfg (dict) – Configuration.

  • logdir (string) – Path to folder where CSV logs can be stored.

  • iteration (int) – Current iteration number.

Notes

N/A.

mlreco.post_processing.metrics.bayes_segnet_mcdropout module mlreco.post_processing.metrics.cluster_gnn_metrics module