diff --git a/RobustParTAK4/PUPPI/V00/modelfile b/RobustParTAK4/PUPPI/V00/modelfile new file mode 120000 index 0000000..7f65391 --- /dev/null +++ b/RobustParTAK4/PUPPI/V00/modelfile @@ -0,0 +1 @@ +../../../models/particletransformer_AK4/1/ \ No newline at end of file diff --git a/UParTAK4/PUPPI/V00/modelfile b/UParTAK4/PUPPI/V00/modelfile new file mode 120000 index 0000000..caf71de --- /dev/null +++ b/UParTAK4/PUPPI/V00/modelfile @@ -0,0 +1 @@ +../../../models/unifiedparticletransformer_AK4/1/ \ No newline at end of file diff --git a/RobustParTAK4/PUPPI/V00/RobustParTAK4.onnx b/models/particletransformer_AK4/1/model.onnx similarity index 100% rename from RobustParTAK4/PUPPI/V00/RobustParTAK4.onnx rename to models/particletransformer_AK4/1/model.onnx diff --git a/models/particletransformer_AK4/config.pbtxt b/models/particletransformer_AK4/config.pbtxt new file mode 100644 index 0000000..e007fdd --- /dev/null +++ b/models/particletransformer_AK4/config.pbtxt @@ -0,0 +1,47 @@ +name: "particletransformer_AK4" +platform: "onnxruntime_onnx" +max_batch_size : 500 +dynamic_batching { + preferred_batch_size: [ 200 ] +} +input [ + { + name: "input_1" + data_type: TYPE_FP32 + dims: [ -1, 16 ] + }, + { + name: "input_2" + data_type: TYPE_FP32 + dims: [ -1, 8 ] + }, + { + name: "input_3" + data_type: TYPE_FP32 + dims: [ -1, 14 ] + }, + { + name: "input_4" + data_type: TYPE_FP32 + dims: [ -1, 4 ] + }, + { + name: "input_5" + data_type: TYPE_FP32 + dims: [ -1, 4 ] + }, + { + name: "input_6" + data_type: TYPE_FP32 + dims: [ -1, 4 ] + } +] +output [ + { + name: "softmax" + data_type: TYPE_FP32 + dims: [ 6 ] + label_filename: "particletransformer_labels.txt" + } +] +optimization {graph : {level : -1}} diff --git a/models/particletransformer_AK4/particletransformer_labels.txt b/models/particletransformer_AK4/particletransformer_labels.txt new file mode 100644 index 0000000..e4068c0 --- /dev/null +++ b/models/particletransformer_AK4/particletransformer_labels.txt @@ -0,0 +1,6 @@ +b +bb +lepb +c +uds +g diff --git a/UParTAK4/PUPPI/V00/UParTAK4.onnx b/models/unifiedparticletransformer_AK4/1/model.onnx similarity index 98% rename from UParTAK4/PUPPI/V00/UParTAK4.onnx rename to models/unifiedparticletransformer_AK4/1/model.onnx index 05fed03..b76de3c 100644 Binary files a/UParTAK4/PUPPI/V00/UParTAK4.onnx and b/models/unifiedparticletransformer_AK4/1/model.onnx differ diff --git a/models/unifiedparticletransformer_AK4/config.pbtxt b/models/unifiedparticletransformer_AK4/config.pbtxt new file mode 100644 index 0000000..dddea1c --- /dev/null +++ b/models/unifiedparticletransformer_AK4/config.pbtxt @@ -0,0 +1,53 @@ +name: "unifiedparticletransformer_AK4" +platform: "onnxruntime_onnx" +input [ + { + name: "input_1" + data_type: TYPE_FP32 + dims: [ -1, -1, 25 ] + }, + { + name: "input_2" + data_type: TYPE_FP32 + dims: [ -1, -1, 18 ] + }, + { + name: "input_3" + data_type: TYPE_FP32 + dims: [ -1, -1, 8 ] + }, + { + name: "input_4" + data_type: TYPE_FP32 + dims: [ -1, -1, 14 ] + }, + { + name: "input_5" + data_type: TYPE_FP32 + dims: [ -1, -1, 4 ] + }, + { + name: "input_6" + data_type: TYPE_FP32 + dims: [ -1, -1, 4 ] + }, + { + name: "input_7" + data_type: TYPE_FP32 + dims: [ -1, -1, 4 ] + }, + { + name: "input_8" + data_type: TYPE_FP32 + dims: [ -1, -1, 4 ] + } +] +output [ + { + name: "softmax" + data_type: TYPE_FP32 + dims: [ -1, 30 ] + label_filename: "unifiedparticletransformer_labels.txt" + } +] +optimization {graph : {level : -1}} diff --git a/models/unifiedparticletransformer_AK4/unifiedparticletransformer_labels.txt b/models/unifiedparticletransformer_AK4/unifiedparticletransformer_labels.txt new file mode 100644 index 0000000..8bae45e --- /dev/null +++ b/models/unifiedparticletransformer_AK4/unifiedparticletransformer_labels.txt @@ -0,0 +1,30 @@ +b +bb +lepb +c +s +u +d +g +ele +mu +taup1h0p +taup1h1p +taup1h2p +taup3h0p +taup3h1p +taum1h0p +taum1h1p +taum1h2p +taum3h0p +taum3h1p +ptcorr +ptreshigh +ptreslow +ptnu +emudata +emumc +dimudata +dimumc +mutaudata +mutaumc