Functions
	main (args)

Variables
	parser = SetupArgParser()
	nEvents
	inputFile
	required
	True
	dest
	help
	type
	float
	default
	int
	action
	score_mode = parser.add_mutually_exclusive_group()
	choices
	DEFAULT_BUCKET_SINGLE_OUTPUT_MODE
	const
	False
	args = parser.parse_args()

Detailed Description

Run segment-edge ONNX inference with an optional JSONL parity dump.

Function Documentation

◆ main()

muonEdgeSegmentInference.main ( args )

Definition at line 6 of file muonEdgeSegmentInference.py.

def main(args):
    from MuonGeoModelTestR4.testGeoModel import setupGeoR4TestCfg
    from MuonConfig.MuonConfigUtils import executeTest
    from AthenaConfiguration.AllConfigFlags import initConfigFlags
    flags = initConfigFlags()
 
    if args.athenaDebug:
        flags.Exec.DebugMessageComponents = [
            "SegmentEdgeInferenceAlg",
            "SegmentEdgeInferenceAlg.SegmentEdgeClassifierTool",
            "SegmentEdgeInferenceAlg.SegmentTrackCandidateBuilderTool",
            "SegmentEdgeInferenceAlg.SegmentEdgeClassifierTool.OnnxRuntimeSessionToolCPU",
            "SegmentEdgeInferenceAlg.SegmentEdgeClassifierTool.OnnxRuntimeSessionToolCUDA",
            "GraphInferenceAlg",
            "GraphBucketFilterTool",
        ]
        print("INFO: Exec.DebugMessageComponents configured:", flags.Exec.DebugMessageComponents)
 
    from AthOnnxComps.OnnxRuntimeFlags import OnnxRuntimeType
    if args.use_cpu:
        flags.AthOnnx.ExecutionProvider = OnnxRuntimeType.CPU
    else:
        flags.AthOnnx.ExecutionProvider = OnnxRuntimeType.CUDA
 
    flags, cfg = setupGeoR4TestCfg(args, flags)
 
    from MuonConfig.MuonDataPrepConfig import xAODUncalibMeasPrepCfg
    cfg.merge(xAODUncalibMeasPrepCfg(flags))
 
    from MuonSpacePointFormation.SpacePointFormationConfig import MuonSpacePointFormationCfg
    cfg.merge(MuonSpacePointFormationCfg(flags))
 
    from MuonPatternRecognitionAlgs.MuonPatternRecognitionConfig import MuonPatternRecognitionCfg
    if args.doMLBucketFilter:
        from MuonInference.InferenceConfig import GraphBucketFilterToolCfg, GraphInferenceAlgCfg
        bucket_tool = cfg.popToolsAndMerge(
            GraphBucketFilterToolCfg(
                flags,
                ModelPath=args.bucket_model_path,
                ScoreThreshold=args.score_threshold,
                OutputName=args.output_name,
                SingleOutputMode=args.single_output_mode,
            )
        )
        cfg.merge(GraphInferenceAlgCfg(flags, InferenceTools=[bucket_tool]))
        cfg.merge(MuonPatternRecognitionCfg(flags))
        cfg.getEventAlgo("MuonEtaHoughTransformAlg").SpacePointContainer = "FilteredMlBuckets"
    else:
        cfg.merge(MuonPatternRecognitionCfg(flags))
 
    output_level = 1 if args.athenaDebug else 3
    edge_space_point_key = "FilteredMlBuckets" if args.doMLBucketFilter else "MuonSpacePoints"
 
    edge_classifier_kwargs = {
        "ModelPath": args.edgeModel,
        "ReadSpacePoints": edge_space_point_key,
        "DebugDumpFile": args.segment_edge_debug_dump_file,
        "DebugDumpMaxEvents": args.segment_edge_debug_dump_max_events,
        "MaxDeltaThetaDeg": args.max_delta_theta_deg,
        "MaxDeltaSector": args.max_delta_sector,
        "SectorModulo": args.sector_modulo,
        "OutputLevel": output_level,
    }
    from MuonInference.InferenceConfig import SegmentEdgeInferenceAlgCfg
    cfg.merge(
        SegmentEdgeInferenceAlgCfg(
            flags,
            EdgeClassifierTool=edge_classifier_kwargs,
            EdgeThreshold=args.edge_threshold,
            OutputLevel=output_level,
        )
    )
 
    executeTest(cfg)
 