add autodetection

Azaion.Annotator/DTO/Config.cs

@@ -37,7 +37,15 @@ public class Config
 
     public ThumbnailConfig ThumbnailConfig { get; set; } = null!;
     public int? LastSelectedExplorerClass { get; set; }
+    public AIRecognitionConfig AIRecognitionConfig { get; set; } = null!;
+}
+
+public class AIRecognitionConfig
+{
     public string AIModelPath { get; set; } = null!;
+    public double FrameRecognitionSeconds { get; set; }
+    public double TrackingDistanceConfidence { get; set; }
+    public double TrackingProbabilityIncrease { get; set; }
 }
 
 public class WindowConfig
@@ -71,6 +79,11 @@ public class FileConfigRepository(ILogger<FileConfigRepository> logger) : IConfi
     private const string DEFAULT_THUMBNAILS_DIR = "thumbnails";
     private const string DEFAULT_UNKNOWN_IMG_DIR = "unknown";
 
+    private const int DEFAULT_THUMBNAIL_BORDER = 10;
+    private const double DEFAULT_FRAME_RECOGNITION_SECONDS = 2;
+    private const double TRACKING_DISTANCE_CONFIDENCE = 0.15;
+    private const double TRACKING_PROBABILITY_INCREASE = 15;
+
     private static readonly Size DefaultWindowSize = new(1280, 720);
     private static readonly Point DefaultWindowLocation = new(100, 100);
     private static readonly Size DefaultThumbnailSize = new(240, 135);
@@ -111,7 +124,14 @@ public class FileConfigRepository(ILogger<FileConfigRepository> logger) : IConfi
                 ThumbnailConfig = new ThumbnailConfig
                 {
                     Size = DefaultThumbnailSize,
-                    Border = 10
+                    Border = DEFAULT_THUMBNAIL_BORDER
+                },
+                AIRecognitionConfig = new AIRecognitionConfig
+                {
+                    AIModelPath = "azaion.onnx",
+                    FrameRecognitionSeconds = DEFAULT_FRAME_RECOGNITION_SECONDS,
+                    TrackingDistanceConfidence = TRACKING_DISTANCE_CONFIDENCE,
+                    TrackingProbabilityIncrease = TRACKING_PROBABILITY_INCREASE
                 }
             };
         }

Azaion.Annotator/DTO/Label.cs

@@ -9,11 +9,11 @@ public abstract class Label
 {
     [JsonProperty(PropertyName = "cl")] public int ClassNumber { get; set; }
 
-    public Label()
+    protected Label()
     {
     }
 
-    public Label(int classNumber)
+    protected Label(int classNumber)
     {
         ClassNumber = classNumber;
     }
@@ -25,20 +25,22 @@ public class CanvasLabel : Label
     public double Y { get; set; }
     public double Width { get; set; }
     public double Height { get; set; }
+    public double? Probability { get; }
 
     public CanvasLabel()
     {
     }
 
-    public CanvasLabel(int classNumber, double x, double y, double width, double height) : base(classNumber)
+    public CanvasLabel(int classNumber, double x, double y, double width, double height, double? probability = null) : base(classNumber)
     {
         X = x;
         Y = y;
         Width = width;
         Height = height;
+        Probability = probability;
     }
 
-    public CanvasLabel(YoloLabel label, Size canvasSize, Size videoSize)
+    public CanvasLabel(YoloLabel label, Size canvasSize, Size videoSize, double? probability = null)
     {
         var cw = canvasSize.Width;
         var ch = canvasSize.Height;
@@ -70,6 +72,7 @@ public class CanvasLabel : Label
             Width = label.Width * realWidth;
             Height = label.Height * ch;
         }
+        Probability = probability;
     }
 }
 
@@ -158,10 +161,10 @@ public class YoloLabel : Label
             .ToList()!;
     }
 
-    public static async Task WriteToFile(IEnumerable<YoloLabel> labels, string filename)
+    public static async Task WriteToFile(IEnumerable<YoloLabel> labels, string filename, CancellationToken cancellationToken = default)
     {
         var labelsStr = string.Join(Environment.NewLine, labels.Select(x => x.ToString()));
-        await File.WriteAllTextAsync(filename, labelsStr);
+        await File.WriteAllTextAsync(filename, labelsStr, cancellationToken);
     }
 
     public override string ToString() => $"{ClassNumber} {CenterX:F5} {CenterY:F5} {Width:F5} {Height:F5}".Replace(',', '.');

Azaion.Annotator/DTO/MediatrEvents.cs

@@ -19,4 +19,3 @@ public class VolumeChangedEvent(int volume) : INotification
     public int Volume { get; set; } = volume;
 }
 
-public class AIDetectEvent : IRequest<List<YoloLabel>>;