File indexing completed on 2025-03-14 23:35:56
0001 import FWCore.ParameterSet.Config as cms
0002
0003 def TracksterLinksProducer(*args, **kwargs):
0004 mod = cms.EDProducer('TracksterLinksProducer',
0005 pluginInferenceAlgoTracksterInferenceByDNN = cms.PSet(
0006 algo_verbosity = cms.int32(0),
0007 onnxPIDModelPath = cms.FileInPath('RecoHGCal/TICL/data/ticlv5/onnx_models/DNN/patternrecognition/id_v0.onnx'),
0008 onnxEnergyModelPath = cms.FileInPath('RecoHGCal/TICL/data/ticlv5/onnx_models/DNN/patternrecognition/energy_v0.onnx'),
0009 inputNames = cms.vstring('input'),
0010 output_en = cms.vstring('enreg_output'),
0011 output_id = cms.vstring('pid_output'),
0012 eid_min_cluster_energy = cms.double(1),
0013 eid_n_layers = cms.int32(50),
0014 eid_n_clusters = cms.int32(10),
0015 doPID = cms.int32(1),
0016 doRegression = cms.int32(1),
0017 type = cms.string('TracksterInferenceByDNN')
0018
0019 ),
0020 pluginInferenceAlgoTracksterInferenceByPFN = cms.PSet(
0021 algo_verbosity = cms.int32(0),
0022 onnxPIDModelPath = cms.FileInPath('RecoHGCal/TICL/data/ticlv5/onnx_models/PFN/patternrecognition/id_v0.onnx'),
0023 onnxEnergyModelPath = cms.FileInPath('RecoHGCal/TICL/data/ticlv5/onnx_models/PFN/patternrecognition/energy_v0.onnx'),
0024 inputNames = cms.vstring(
0025 'input',
0026 'input_tr_features'
0027 ),
0028 output_en = cms.vstring('enreg_output'),
0029 output_id = cms.vstring('pid_output'),
0030 eid_min_cluster_energy = cms.double(1),
0031 eid_n_layers = cms.int32(50),
0032 eid_n_clusters = cms.int32(10),
0033 doPID = cms.int32(1),
0034 doRegression = cms.int32(1),
0035 type = cms.string('TracksterInferenceByPFN')
0036
0037 ),
0038 pluginInferenceAlgoTracksterInferenceByCNNv4 = cms.PSet(
0039 algo_verbosity = cms.int32(0),
0040 onnxModelPath = cms.FileInPath('RecoHGCal/TICL/data/ticlv4/onnx_models/energy_id_v0.onnx'),
0041 inputNames = cms.vstring('input:0'),
0042 outputNames = cms.vstring(
0043 'output/regressed_energy:0',
0044 'output/id_probabilities:0'
0045 ),
0046 eid_min_cluster_energy = cms.double(1),
0047 eid_n_layers = cms.int32(50),
0048 eid_n_clusters = cms.int32(10),
0049 doPID = cms.int32(1),
0050 doRegression = cms.int32(0),
0051 type = cms.string('TracksterInferenceByCNNv4')
0052
0053 ),
0054 linkingPSet = cms.PSet(
0055 cylinder_radius_sqr_split = cms.double(9),
0056 proj_distance_split = cms.double(5),
0057 track_time_quality_threshold = cms.double(0.5),
0058 min_num_lcs = cms.uint32(7),
0059 min_trackster_energy = cms.double(10),
0060 pca_quality_th = cms.double(0.85),
0061 dot_prod_th = cms.double(0.97),
0062 deltaRxy = cms.double(4),
0063 lower_boundary = cms.vdouble(
0064 10,
0065 150
0066 ),
0067 upper_boundary = cms.vdouble(
0068 3,
0069 70
0070 ),
0071 upper_distance_projective_sqr = cms.vdouble(
0072 40,
0073 60
0074 ),
0075 lower_distance_projective_sqr = cms.vdouble(
0076 10,
0077 30
0078 ),
0079 min_distance_z = cms.vdouble(
0080 35,
0081 35
0082 ),
0083 upper_distance_projective_sqr_closest_points = cms.vdouble(
0084 10,
0085 30
0086 ),
0087 lower_distance_projective_sqr_closest_points = cms.vdouble(
0088 10,
0089 30
0090 ),
0091 max_z_distance_closest_points = cms.vdouble(
0092 35,
0093 35
0094 ),
0095 cylinder_radius_sqr = cms.vdouble(
0096 10,
0097 20
0098 ),
0099 algo_verbosity = cms.int32(0),
0100 type = cms.string('Skeletons')
0101
0102 ),
0103 tracksters_collections = cms.VInputTag('ticlTrackstersCLUE3DHigh'),
0104 original_masks = cms.VInputTag('hgcalMergeLayerClusters:InitialLayerClustersMask'),
0105 layer_clusters = cms.InputTag('hgcalMergeLayerClusters'),
0106 layer_clustersTime = cms.InputTag('hgcalMergeLayerClusters', 'timeLayerCluster'),
0107 regressionAndPid = cms.bool(False),
0108 detector = cms.string('HGCAL'),
0109 propagator = cms.string('PropagatorWithMaterial'),
0110 inferenceAlgo = cms.string('TracksterInferenceByPFN'),
0111 mightGet = cms.optional.untracked.vstring
0112 )
0113 for a in args:
0114 mod.update_(a)
0115 mod.update_(kwargs)
0116 return mod
