Bugfix found during regression

TensorStack-AI · May 22, 2024 · e17e737 · e17e737
1 parent 8993731
commit e17e737
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 4 deletions.
diff --git a/OnnxStack.Core/Config/OnnxModelPrecision.cs b/OnnxStack.Core/Config/OnnxModelPrecision.cs
@@ -2,7 +2,7 @@
 {
     public enum OnnxModelPrecision
     {
-        F32 = 0,
-        F16 = 1
+        F16 = 0,
+        F32 = 1
     }
 }
diff --git a/OnnxStack.Core/Video/VideoInfo.cs b/OnnxStack.Core/Video/VideoInfo.cs
@@ -84,7 +84,11 @@ private float GetFramesPerSecond()
             if (FrameCount == 0 || DurationSeconds == 0)
                 return 0;
 
-            return FrameCount / DurationSeconds;
+            var framesPerSec = FrameCount / DurationSeconds;
+            if (framesPerSec < 1)
+                return MathF.Round(framesPerSec, 2);
+
+            return MathF.Round(framesPerSec, 0);
         }
     }
 }
diff --git a/OnnxStack.FeatureExtractor/Pipelines/FeatureExtractorPipeline.cs b/OnnxStack.FeatureExtractor/Pipelines/FeatureExtractorPipeline.cs
@@ -102,10 +102,11 @@ public async Task<OnnxVideo> RunAsync(OnnxVideo video, Action<OnnxImage, OnnxIma
             var featureFrames = new List<OnnxImage>();
             foreach (var videoFrame in video.Frames)
             {
-                var result = await RunAsync(videoFrame, cancellationToken);
+                var result = await ExtractImageAsync(videoFrame, cancellationToken);
                 featureFrames.Add(result);
                 progressCallback?.Invoke(videoFrame, result);
             }
+
             _logger?.LogEnd("Extracting OnnxVideo features complete.", timestamp);
             return new OnnxVideo(video.Info, featureFrames);
         }