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.