feat: refactor to proper app + write tests

2025-02-13 17:24:38 -06:00
parent baf4035616
commit 42435ad915
38 changed files with 1462 additions and 253 deletions
@@ -0,0 +1,18 @@
+namespace BGR.Console.Removal.Onnx;
+
+internal class OnnxInferenceRunner : IInferenceRunner
+{
+  public ITensor<float> Run(byte[] model, ITensor<float> inputTensor)
+  {
+    using var options = new SessionOptions() { LogSeverityLevel = OrtLoggingLevel.ORT_LOGGING_LEVEL_ERROR };
+    using var session = new InferenceSession(model, options);
+    var inputs = new List<NamedOnnxValue>()
+    {
+      NamedOnnxValue.CreateFromTensor(session.InputNames[0], inputTensor.ToTensor()),
+    };
+
+    var results = session.Run(inputs);
+    var outputTensor = results[0].AsTensor<float>();
+    return new OnnxTensor(outputTensor);
+  }
+}