Add timer

BobLd · Jul 17, 2021 · 65674c3 · 65674c3
1 parent abd33a2
commit 65674c3
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/YOLOv4MLNet/Program.cs b/YOLOv4MLNet/Program.cs
@@ -1,5 +1,7 @@
 using Microsoft.ML;
+using System;
 using System.Collections.Generic;
+using System.Diagnostics;
 using System.Drawing;
 using System.IO;
 using YOLOv4MLNet.DataStructures;
@@ -59,8 +61,9 @@ static void Main()
 
             // save model
             //mlContext.Model.Save(model, predictionEngine.OutputSchema, Path.ChangeExtension(modelPath, "zip"));
-
-            foreach (string imageName in new string[] { "kite.jpg", "kite_416.jpg", "dog_cat.jpg", "cars road.jpg", "ski.jpg", "ski2.jpg" })
+            var sw = new Stopwatch();
+            sw.Start();
+            foreach (string imageName in new string[] { "kite.jpg", "dog_cat.jpg", "cars road.jpg", "ski.jpg", "ski2.jpg" })
             {
                 using (var bitmap = new Bitmap(Image.FromFile(Path.Combine(imageFolder, imageName))))
                 {
@@ -90,6 +93,8 @@ static void Main()
                     }
                 }
             }
+            sw.Stop();
+            Console.WriteLine($"Done in {sw.ElapsedMilliseconds}ms.");
         }
     }
 }