mlreco.post_processing.analysis.track_clustering module

mlreco.post_processing.analysis.track_clustering.track_clustering(cfg, data_blob, res, logdir, iteration)[source]

Track clustering on PPN+UResNet output.

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

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

  • cfg (dict) – Configuration.

  • debug (bool, optional) – Whether to print some stats or not in the stdout.

Notes

Based on - semantic segmentation output - point position and type predictions In addition, includes a break algorithm to break track clusters into smaller clusters based on predicted points. Stores all points and informations in a CSV file.