Skip to content

Add onnx metadata executable

Samuel Van Stroud requested to merge (removed):master-OnnxMetadataExe into master

A simple CLI tool to read metadata from an ONNX model.

Example:

get-onnx-metadata /cvmfs/atlas.cern.ch/repo/sw/database/GroupData/dev/BTagging/20220420/GN1v00/antikt4empflow/network.onnx gnn_config
{"input_sequences": [{"name": "tracks_dipsLoose202102_sd0sort", "variables": [{"name": "d0", "offset": 0.0, "scale": 1.0}, {"name": "z0SinTheta", "offset": 0.0, "scale": 1.0}, {"name": "dphi", "offset": 0.0, "scale": 1.0}, {"name": "deta", "offset": 0.0, "scale": 1.0}, {"name": "qOverP", "offset": 0.0, "scale": 1.0}, {"name": "IP3D_signed_d0_significance", "offset": 0.0, "scale": 1.0}, {"name": "IP3D_signed_z0_significance", "offset": 0.0, "scale": 1.0}, {"name": "phiUncertainty", "offset": 0.0, "scale": 1.0}, {"name": "thetaUncertainty", "offset": 0.0, "scale": 1.0}, {"name": "qOverPUncertainty", "offset": 0.0, "scale": 1.0}, {"name": "numberOfPixelHits", "offset": 0.0, "scale": 1.0}, {"name": "numberOfSCTHits", "offset": 0.0, "scale": 1.0}, {"name": "numberOfInnermostPixelLayerHits", "offset": 0.0, "scale": 1.0}, {"name": "numberOfNextToInnermostPixelLayerHits", "offset": 0.0, "scale": 1.0}, {"name": "numberOfInnermostPixelLayerSharedHits", "offset": 0.0, "scale": 1.0}, {"name": "numberOfInnermostPixelLayerSplitHits", "offset": 0.0, "scale": 1.0}, {"name": "numberOfPixelSharedHits", "offset": 0.0, "scale": 1.0}, {"name": "numberOfPixelSplitHits", "offset": 0.0, "scale": 1.0}, {"name": "numberOfSCTSharedHits", "offset": 0.0, "scale": 1.0}, {"name": "numberOfPixelHoles", "offset": 0.0, "scale": 1.0}, {"name": "numberOfSCTHoles", "offset": 0.0, "scale": 1.0}]}], "inputs": [{"name": "jet_var", "variables": [{"name": "eta", "offset": 0.0, "scale": 1.0}, {"name": "pt", "offset": 0.0, "scale": 1.0}]}], "layers": [], "nodes": [], "outputs": {"GN1v00": {"labels": ["pu", "pc", "pb"], "node_index": 0}}}

@dguest @mguth @nkakati

Edited by Samuel Van Stroud

Merge request reports