fix ai detection bugs #1

Azaion.Annotator/DTO/Config.cs

@@ -46,6 +46,7 @@ public class AIRecognitionConfig
     public double FrameRecognitionSeconds { get; set; }
     public double TrackingDistanceConfidence { get; set; }
     public double TrackingProbabilityIncrease { get; set; }
+    public double TrackingIntersectionThreshold { get; set; }
 }
 
 public class WindowConfig
@@ -83,6 +84,7 @@ public class FileConfigRepository(ILogger<FileConfigRepository> logger) : IConfi
     private const double DEFAULT_FRAME_RECOGNITION_SECONDS = 2;
     private const double TRACKING_DISTANCE_CONFIDENCE = 0.15;
     private const double TRACKING_PROBABILITY_INCREASE = 15;
+    private const double TRACKING_INTERSECTION_THRESHOLD = 0.8;
 
     private static readonly Size DefaultWindowSize = new(1280, 720);
     private static readonly Point DefaultWindowLocation = new(100, 100);
@@ -131,7 +133,8 @@ public class FileConfigRepository(ILogger<FileConfigRepository> logger) : IConfi
                     AIModelPath = "azaion.onnx",
                     FrameRecognitionSeconds = DEFAULT_FRAME_RECOGNITION_SECONDS,
                     TrackingDistanceConfidence = TRACKING_DISTANCE_CONFIDENCE,
-                    TrackingProbabilityIncrease = TRACKING_PROBABILITY_INCREASE
+                    TrackingProbabilityIncrease = TRACKING_PROBABILITY_INCREASE,
+                    TrackingIntersectionThreshold = TRACKING_INTERSECTION_THRESHOLD
                 }
             };
         }

Azaion.Annotator/DTO/FormState.cs

@@ -15,6 +15,7 @@ public class FormState
     public Size CurrentVideoSize { get; set; }
     public TimeSpan CurrentVideoLength { get; set; }
 
+    public bool BackgroundShown { get; set; }
     public int CurrentVolume { get; set; } = 100;
     public ObservableCollection<AnnotationResult> AnnotationResults { get; set; } = [];
     public WindowsEnum ActiveWindow { get; set; }

Azaion.Annotator/DTO/Label.cs

@@ -1,7 +1,8 @@
-using System.Globalization;
+using System.Drawing;
+using System.Globalization;
 using System.IO;
-using System.Windows;
 using Newtonsoft.Json;
+using Size = System.Windows.Size;
 
 namespace Azaion.Annotator.DTO;
 
@@ -98,6 +99,9 @@ public class YoloLabel : Label
         Height = height;
     }
 
+    public RectangleF ToRectangle() =>
+        new((float)(CenterX - Width / 2.0), (float)(CenterY - Height / 2.0), (float)Width, (float)Height);
+
     public YoloLabel(CanvasLabel canvasLabel, Size canvasSize, Size videoSize)
     {
         var cw = canvasSize.Width;
@@ -168,4 +172,18 @@ public class YoloLabel : Label
     }
 
     public override string ToString() => $"{ClassNumber} {CenterX:F5} {CenterY:F5} {Width:F5} {Height:F5}".Replace(',', '.');
+}
+
+public class Detection : YoloLabel
+{
+    public Detection(YoloLabel label, double? probability = null)
+    {
+        ClassNumber = label.ClassNumber;
+        CenterX = label.CenterX;
+        CenterY = label.CenterY;
+        Height = label.Height;
+        Width = label.Width;
+        Probability = probability;
+    }
+    public double? Probability { get; set; }
 }