C# YoloV8 OpenVINO 视频抽帧 自动标注 预标注工具

C# YoloV8 OpenVINO 视频抽帧 自动标注 预标注工具

目录

效果

项目

代码

下载 

效果

视频抽帧 自动标注 预标注工具 效果

项目

代码

using OpenCvSharp;
using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.IO;
using System.Text;
using System.Threading;
using System.Threading.Tasks;
using System.Windows.Forms;

namespace yolov8_OpenVINO_Demo
{
    public partial class Form2 : Form
    {
        public Form2()
        {
            InitializeComponent();
        }

        YoloV8 yoloV8;
        YoloV8Async yoloV8Async;

        string model_path;

        string video_path = "";
        string video_name = "";
        string videoFilter = "*.mp4|*.mp4;";
        VideoCapture vcapture;

        string output_path = "";
        string images_path = "";
        string labels_path = "";


        StringBuilder sb = new StringBuilder();

        ///


        /// 窗体加载，初始化
        ///
        ///
        ///
        private void Form1_Load(object sender, EventArgs e)
        {
            model_path = "model/yolov8n.onnx";
            yoloV8 = new YoloV8(model_path, "model/lable.txt");

            yoloV8Async = new YoloV8Async(model_path, "model/lable.txt");
        }

        ///


        /// 选择输出目录
        ///
        ///
        ///
        private void button2_Click(object sender, EventArgs e)
        {
            output_path = "";
            using (FolderBrowserDialog folderBrowserDialog = new FolderBrowserDialog())
            {
                folderBrowserDialog.Description = "选择目录";
                DialogResult dialogResult = folderBrowserDialog.ShowDialog();
                if (dialogResult == DialogResult.OK)
                {
                    output_path = folderBrowserDialog.SelectedPath;

                }
            }
        }

        ///


        /// 选择视频
        ///
        ///
        ///
        private void button4_Click(object sender, EventArgs e)
        {
            OpenFileDialog ofd = new OpenFileDialog();
            ofd.Filter = videoFilter;
            ofd.InitialDirectory = Application.StartupPath + "\\test";
            if (ofd.ShowDialog() != DialogResult.OK) return;

            video_path = ofd.FileName;
            video_name = System.IO.Path.GetFileNameWithoutExtension(video_path);
            textBox1.Text = "";

        }

        ///


        /// 同步接口-视频推理
        ///
        ///
        ///
        private void button3_Click(object sender, EventArgs e)
        {
            if (video_path == "")
            {
                MessageBox.Show("请先选择视频！");
                return;
            }

            if (output_path == "")
            {
                MessageBox.Show("请先选择输出目录！");
                return;
            }

            images_path = output_path + "\\output\\images";
            labels_path = output_path + "\\output\\labels";

            if (!Directory.Exists(images_path))
            {
                Directory.CreateDirectory(images_path);
            }

            if (!Directory.Exists(labels_path))
            {
                Directory.CreateDirectory(labels_path);
            }

            sb.Clear();
            sb.AppendLine($"images path:{images_path}");
            sb.AppendLine($"labels path:{labels_path}");

            textBox1.Text = sb.ToString();

            Application.DoEvents();

            Thread thread = new Thread(new ThreadStart(VideoDetection));

            thread.Start();
            thread.Join();

            textBox1.Text = "检测完成！";
        }

        void VideoDetection()
        {
            vcapture = new VideoCapture(video_path);
            if (!vcapture.IsOpened())
            {
                MessageBox.Show("打开视频文件失败");
                return;
            }

            Mat frame = new Mat();
            List detResults;

            // 获取视频的fps
            double videoFps = vcapture.Get(VideoCaptureProperties.Fps);
            // 计算等待时间（毫秒）
            int delay = (int)(1000 / videoFps);
            Stopwatch _stopwatch = new Stopwatch();

            int frameCount = (int)vcapture.Get(VideoCaptureProperties.FrameCount);

            Cv2.NamedWindow("DetectionResult 按下ESC，退出", WindowFlags.Normal);
            Cv2.ResizeWindow("DetectionResult 按下ESC，退出", vcapture.FrameWidth / 2, vcapture.FrameHeight / 2);

            int FrameWidth = vcapture.FrameWidth;
            int FrameHeight = vcapture.FrameHeight;
            int index = 0;

            while (vcapture.Read(frame))
            {
                if (frame.Empty())
                {
                    MessageBox.Show("读取失败");
                    return;
                }

                _stopwatch.Restart();

                delay = (int)(1000 / videoFps);

                detResults = yoloV8.Detect(frame);

                //输出图片
                Task.Factory.StartNew(() =>
                {
                    Cv2.ImWrite(images_path + "\\" + video_name + "_" + index.ToString() + ".jpg", frame);
                });

                StringBuilder sbLabels = new StringBuilder();
                //绘制结果
                foreach (DetectionResult r in detResults)
                {
                    Cv2.PutText(frame, $"{r.Class}:{r.Confidence:P0}", new OpenCvSharp.Point(r.Rect.TopLeft.X, r.Rect.TopLeft.Y - 10), HersheyFonts.HersheySimplex, 1, Scalar.Red, 2);
                    Cv2.Rectangle(frame, r.Rect, Scalar.Red, thickness: 2);

                    //class, x_center, y_center, width, height
                    double width = Math.Round(r.Rect.Width / (double)FrameWidth, 6);
                    double height = Math.Round(r.Rect.Height / (double)FrameHeight, 6);
                    double x_center = Math.Round((r.Rect.Right - (r.Rect.Width / 2)) / (double)FrameWidth, 6);
                    double y_center = Math.Round((r.Rect.Bottom - (r.Rect.Height / 2)) / (double)FrameHeight, 6);

                    sbLabels.Append($"{r.ClassId} {x_center} {y_center} {width} {height}\n");
                }

                //生成不带BOM格式
                using (FileStream fs = new FileStream(labels_path + "\\" + video_name + "_" + index.ToString() + ".txt", FileMode.Create))
                {
                    using (StreamWriter sw = new StreamWriter(fs, new UTF8Encoding(false)))
                    {
                        sw.Write(sbLabels.ToString());
                    }
                }
                //Cv2.ImWrite(images_path + "\\" + index.ToString() + ".jpg", frame);
                index++;

                Cv2.PutText(frame, "preprocessTime:" + yoloV8.preprocessTime.ToString("F2") + "ms", new OpenCvSharp.Point(10, 30), HersheyFonts.HersheySimplex, 1, Scalar.Red, 2);
                Cv2.PutText(frame, "inferTime:" + yoloV8.inferTime.ToString("F2") + "ms", new OpenCvSharp.Point(10, 70), HersheyFonts.HersheySimplex, 1, Scalar.Red, 2);
                Cv2.PutText(frame, "postprocessTime:" + yoloV8.postprocessTime.ToString("F2") + "ms", new OpenCvSharp.Point(10, 110), HersheyFonts.HersheySimplex, 1, Scalar.Red, 2);
                Cv2.PutText(frame, "totalTime:" + yoloV8.totalTime.ToString("F2") + "ms", new OpenCvSharp.Point(10, 150), HersheyFonts.HersheySimplex, 1, Scalar.Red, 2);
                Cv2.PutText(frame, "video fps:" + videoFps.ToString("F2"), new OpenCvSharp.Point(10, 190), HersheyFonts.HersheySimplex, 1, Scalar.Red, 2);
                Cv2.PutText(frame, "det fps:" + yoloV8.detFps.ToString("F2"), new OpenCvSharp.Point(10, 230), HersheyFonts.HersheySimplex, 1, Scalar.Red, 2);
                Cv2.PutText(frame, "progress:" + index.ToString() + "/" + frameCount.ToString(), new OpenCvSharp.Point(10, 270), HersheyFonts.HersheySimplex, 1, Scalar.Red, 2);

                Cv2.ImShow("DetectionResult 按下ESC，退出", frame);

                //for test
                delay = 1;
                //delay = (int)(delay - _stopwatch.ElapsedMilliseconds);
                //if (delay <= 0)
                //{
                //    delay = 1;
                //}
                //Console.WriteLine("delay:" + delay.ToString()) ;
                if (Cv2.WaitKey(delay) == 27 || Cv2.GetWindowProperty("DetectionResult 按下ESC，退出", WindowPropertyFlags.Visible) < 1.0)
                {
                    Cv2.DestroyAllWindows();
                    vcapture.Release();
                    break; // 如果按下ESC，退出循环
                }
            }

            Cv2.DestroyAllWindows();
            vcapture.Release();
        }

        ///


        /// 异步接口-视频推理
        ///
        ///
        ///
        private void button1_Click(object sender, EventArgs e)
        {
            if (video_path == "")
            {
                MessageBox.Show("请先选择视频！");
                return;
            }

            if (output_path == "")
            {
                MessageBox.Show("请先选择输出目录！");
                return;
            }

            images_path = output_path + "\\output\\images";
            labels_path = output_path + "\\output\\labels";

            if (!Directory.Exists(images_path))
            {
                Directory.CreateDirectory(images_path);
            }

            if (!Directory.Exists(labels_path))
            {
                Directory.CreateDirectory(labels_path);
            }

            textBox1.Text = "开始异步推理检测";

            Application.DoEvents();

            Thread thread = new Thread(new ThreadStart(VideoDetectionAsync));

            thread.Start();
            thread.Join();

            textBox1.Text = "异步推理检测完成！";
        }

        void VideoDetectionAsync()
        {
            vcapture = new VideoCapture(video_path);
            if (!vcapture.IsOpened())
            {
                MessageBox.Show("打开视频文件失败");
                return;
            }

            Mat frame = new Mat();
            List detResults;

            // 获取视频的fps
            double videoFps = vcapture.Get(VideoCaptureProperties.Fps);
            // 计算等待时间（毫秒）
            int delay = (int)(1000 / videoFps);
            Stopwatch _stopwatch = new Stopwatch();

            int frameCount = (int)vcapture.Get(VideoCaptureProperties.FrameCount);

            Cv2.NamedWindow("DetectionResult 按下ESC，退出", WindowFlags.Normal);
            Cv2.ResizeWindow("DetectionResult 按下ESC，退出", vcapture.FrameWidth / 2, vcapture.FrameHeight / 2);

            int FrameWidth = vcapture.FrameWidth;
            int FrameHeight = vcapture.FrameHeight;
            int index = 0;

            while (vcapture.Read(frame))
            {
                if (frame.Empty())
                {
                    MessageBox.Show("读取失败");
                    return;
                }

                //输出图片
                Task.Factory.StartNew(() =>
                {
                    Cv2.ImWrite(images_path + "\\" + video_name + "_" + index.ToString() + ".jpg", frame);
                });

                detResults = yoloV8Async.Detect(frame);

                StringBuilder sbLabels = new StringBuilder();
                //绘制结果
                foreach (DetectionResult r in detResults)
                {
                    Cv2.PutText(frame, $"{r.Class}:{r.Confidence:P0}", new OpenCvSharp.Point(r.Rect.TopLeft.X, r.Rect.TopLeft.Y - 10), HersheyFonts.HersheySimplex, 1, Scalar.Red, 2);
                    Cv2.Rectangle(frame, r.Rect, Scalar.Red, thickness: 2);

                    //class, x_center, y_center, width, height
                    double width = Math.Round(r.Rect.Width / (double)FrameWidth, 6);
                    double height = Math.Round(r.Rect.Height / (double)FrameHeight, 6);
                    double x_center = Math.Round((r.Rect.Right - (r.Rect.Width / 2)) / (double)FrameWidth, 6);
                    double y_center = Math.Round((r.Rect.Bottom - (r.Rect.Height / 2)) / (double)FrameHeight, 6);

                    sbLabels.Append($"{r.ClassId} {x_center} {y_center} {width} {height}\n");
                }

                //生成不带BOM格式
                using (FileStream fs = new FileStream(labels_path + "\\" + video_name + "_" + index.ToString() + ".txt", FileMode.Create))
                {
                    using (StreamWriter sw = new StreamWriter(fs, new UTF8Encoding(false)))
                    {
                        sw.Write(sbLabels.ToString());
                    }
                }
                //Cv2.ImWrite(images_path + "\\" + index.ToString() + ".jpg", frame);
                index++;

                Cv2.PutText(frame, "preprocessTime:" + yoloV8Async.preprocessTime.ToString("F2") + "ms", new OpenCvSharp.Point(10, 30), HersheyFonts.HersheySimplex, 1, Scalar.Red, 2);
                Cv2.PutText(frame, "inferTime:" + yoloV8Async.inferTime.ToString("F2") + "ms", new OpenCvSharp.Point(10, 70), HersheyFonts.HersheySimplex, 1, Scalar.Red, 2);
                Cv2.PutText(frame, "postprocessTime:" + yoloV8Async.postprocessTime.ToString("F2") + "ms", new OpenCvSharp.Point(10, 110), HersheyFonts.HersheySimplex, 1, Scalar.Red, 2);
                Cv2.PutText(frame, "totalTime:" + yoloV8Async.totalTime.ToString("F2") + "ms", new OpenCvSharp.Point(10, 150), HersheyFonts.HersheySimplex, 1, Scalar.Red, 2);
                Cv2.PutText(frame, "video fps:" + videoFps.ToString("F2"), new OpenCvSharp.Point(10, 190), HersheyFonts.HersheySimplex, 1, Scalar.Red, 2);
                Cv2.PutText(frame, "det fps:" + yoloV8Async.detFps.ToString("F2"), new OpenCvSharp.Point(10, 230), HersheyFonts.HersheySimplex, 1, Scalar.Red, 2);
                Cv2.PutText(frame, "progress:" + index.ToString() + "/" + frameCount.ToString(), new OpenCvSharp.Point(10, 270), HersheyFonts.HersheySimplex, 1, Scalar.Red, 2);

                //Cv2.ImWrite(images_path + "\\" + index.ToString() + ".jpg", frame);

                Cv2.ImShow("DetectionResult 按下ESC，退出", frame);

                // for test
                delay = 1;
                //delay = (int)(delay - _stopwatch.ElapsedMilliseconds);
                //if (delay <= 0)
                //{
                //    delay = 1;
                //}
                //Console.WriteLine("delay:" + delay.ToString()) ;
                if (Cv2.WaitKey(delay) == 27 || Cv2.GetWindowProperty("DetectionResult 按下ESC，退出", WindowPropertyFlags.Visible) < 1.0)
                {
                    Cv2.DestroyAllWindows();
                    vcapture.Release();
                    break; // 如果按下ESC，退出循环
                }
            }
            Cv2.DestroyAllWindows();
            vcapture.Release();
        }
    }
}

  1. using OpenCvSharp;
  2. using System;
  3. using System.Collections.Generic;
  4. using System.Diagnostics;
  5. using System.IO;
  6. using System.Text;
  7. using System.Threading;
  8. using System.Threading.Tasks;
  9. using System.Windows.Forms;
  10.  
  11. namespace yolov8_OpenVINO_Demo
  12. {
  13.     public partial class Form2 : Form
  14.     {
  15.         public Form2()
  16.         {
  17.             InitializeComponent();
  18.         }
  19.  
  20.         YoloV8 yoloV8;
  21.         YoloV8Async yoloV8Async;
  22.  
  23.         string model_path;
  24.  
  25.         string video_path = "";
  26.         string video_name = "";
  27.         string videoFilter = "*.mp4|*.mp4;";
  28.         VideoCapture vcapture;
  29.  
  30.         string output_path = "";
  31.         string images_path = "";
  32.         string labels_path = "";
  33.  
  34.  
  35.         StringBuilder sb = new StringBuilder();
  36.  
  37.         /// <summary>
  38.         /// 窗体加载，初始化
  39.         /// </summary>
  40.         /// <param name="sender"></param>
  41.         /// <param name="e"></param>
  42.         private void Form1_Load(object sender, EventArgs e)
  43.         {
  44.             model_path = "model/yolov8n.onnx";
  45.             yoloV8 = new YoloV8(model_path, "model/lable.txt");
  46.  
  47.             yoloV8Async = new YoloV8Async(model_path, "model/lable.txt");
  48.         }
  49.  
  50.         /// <summary>
  51.         /// 选择输出目录
  52.         /// </summary>
  53.         /// <param name="sender"></param>
  54.         /// <param name="e"></param>
  55.         private void button2_Click(object sender, EventArgs e)
  56.         {
  57.             output_path = "";
  58.             using (FolderBrowserDialog folderBrowserDialog = new FolderBrowserDialog())
  59.             {
  60.                 folderBrowserDialog.Description = "选择目录";
  61.                 DialogResult dialogResult = folderBrowserDialog.ShowDialog();
  62.                 if (dialogResult == DialogResult.OK)
  63.                 {
  64.                     output_path = folderBrowserDialog.SelectedPath;
  65.  
  66.                 }
  67.             }
  68.         }
  69.  
  70.         /// <summary>
  71.         /// 选择视频
  72.         /// </summary>
  73.         /// <param name="sender"></param>
  74.         /// <param name="e"></param>
  75.         private void button4_Click(object sender, EventArgs e)
  76.         {
  77.             OpenFileDialog ofd = new OpenFileDialog();
  78.             ofd.Filter = videoFilter;
  79.             ofd.InitialDirectory = Application.StartupPath + "\\test";
  80.             if (ofd.ShowDialog() != DialogResult.OK) return;
  81.  
  82.             video_path = ofd.FileName;
  83.             video_name = System.IO.Path.GetFileNameWithoutExtension(video_path);
  84.             textBox1.Text = "";
  85.  
  86.         }
  87.  
  88.         /// <summary>
  89.         /// 同步接口-视频推理
  90.         /// </summary>
  91.         /// <param name="sender"></param>
  92.         /// <param name="e"></param>
  93.         private void button3_Click(object sender, EventArgs e)
  94.         {
  95.             if (video_path == "")
  96.             {
  97.                 MessageBox.Show("请先选择视频！");
  98.                 return;
  99.             }
  100.  
  101.             if (output_path == "")
  102.             {
  103.                 MessageBox.Show("请先选择输出目录！");
  104.                 return;
  105.             }
  106.  
  107.             images_path = output_path + "\\output\\images";
  108.             labels_path = output_path + "\\output\\labels";
  109.  
  110.             if (!Directory.Exists(images_path))
  111.             {
  112.                 Directory.CreateDirectory(images_path);
  113.             }
  114.  
  115.             if (!Directory.Exists(labels_path))
  116.             {
  117.                 Directory.CreateDirectory(labels_path);
  118.             }
  119.  
  120.             sb.Clear();
  121.             sb.AppendLine($"images path:{images_path}");
  122.             sb.AppendLine($"labels path:{labels_path}");
  123.  
  124.             textBox1.Text = sb.ToString();
  125.  
  126.             Application.DoEvents();
  127.  
  128.             Thread thread = new Thread(new ThreadStart(VideoDetection));
  129.  
  130.             thread.Start();
  131.             thread.Join();
  132.  
  133.             textBox1.Text = "检测完成！";
  134.         }
  135.  
  136.         void VideoDetection()
  137.         {
  138.             vcapture = new VideoCapture(video_path);
  139.             if (!vcapture.IsOpened())
  140.             {
  141.                 MessageBox.Show("打开视频文件失败");
  142.                 return;
  143.             }
  144.  
  145.             Mat frame = new Mat();
  146.             List<DetectionResult> detResults;
  147.  
  148.             // 获取视频的fps
  149.             double videoFps = vcapture.Get(VideoCaptureProperties.Fps);
  150.             // 计算等待时间（毫秒）
  151.             int delay = (int)(1000 / videoFps);
  152.             Stopwatch _stopwatch = new Stopwatch();
  153.  
  154.             int frameCount = (int)vcapture.Get(VideoCaptureProperties.FrameCount);
  155.  
  156.             Cv2.NamedWindow("DetectionResult 按下ESC，退出", WindowFlags.Normal);
  157.             Cv2.ResizeWindow("DetectionResult 按下ESC，退出", vcapture.FrameWidth / 2, vcapture.FrameHeight / 2);
  158.  
  159.             int FrameWidth = vcapture.FrameWidth;
  160.             int FrameHeight = vcapture.FrameHeight;
  161.             int index = 0;
  162.  
  163.             while (vcapture.Read(frame))
  164.             {
  165.                 if (frame.Empty())
  166.                 {
  167.                     MessageBox.Show("读取失败");
  168.                     return;
  169.                 }
  170.  
  171.                 _stopwatch.Restart();
  172.  
  173.                 delay = (int)(1000 / videoFps);
  174.  
  175.                 detResults = yoloV8.Detect(frame);
  176.  
  177.                 //输出图片
  178.                 Task.Factory.StartNew(() =>
  179.                 {
  180.                     Cv2.ImWrite(images_path + "\\" + video_name + "_" + index.ToString() + ".jpg", frame);
  181.                 });
  182.  
  183.                 StringBuilder sbLabels = new StringBuilder();
  184.                 //绘制结果
  185.                 foreach (DetectionResult r in detResults)
  186.                 {
  187.                     Cv2.PutText(frame, $"{r.Class}:{r.Confidence:P0}", new OpenCvSharp.Point(r.Rect.TopLeft.X, r.Rect.TopLeft.Y - 10), HersheyFonts.HersheySimplex, 1, Scalar.Red, 2);
  188.                     Cv2.Rectangle(frame, r.Rect, Scalar.Red, thickness: 2);
  189.  
  190.                     //class, x_center, y_center, width, height
  191.                     double width = Math.Round(r.Rect.Width / (double)FrameWidth, 6);
  192.                     double height = Math.Round(r.Rect.Height / (double)FrameHeight, 6);
  193.                     double x_center = Math.Round((r.Rect.Right - (r.Rect.Width / 2)) / (double)FrameWidth, 6);
  194.                     double y_center = Math.Round((r.Rect.Bottom - (r.Rect.Height / 2)) / (double)FrameHeight, 6);
  195.  
  196.                     sbLabels.Append($"{r.ClassId} {x_center} {y_center} {width} {height}\n");
  197.                 }
  198.  
  199.                 //生成不带BOM格式
  200.                 using (FileStream fs = new FileStream(labels_path + "\\" + video_name + "_" + index.ToString() + ".txt", FileMode.Create))
  201.                 {
  202.                     using (StreamWriter sw = new StreamWriter(fs, new UTF8Encoding(false)))
  203.                     {
  204.                         sw.Write(sbLabels.ToString());
  205.                     }
  206.                 }
  207.                 //Cv2.ImWrite(images_path + "\\" + index.ToString() + ".jpg", frame);
  208.                 index++;
  209.  
  210.                 Cv2.PutText(frame, "preprocessTime:" + yoloV8.preprocessTime.ToString("F2") + "ms", new OpenCvSharp.Point(10, 30), HersheyFonts.HersheySimplex, 1, Scalar.Red, 2);
  211.                 Cv2.PutText(frame, "inferTime:" + yoloV8.inferTime.ToString("F2") + "ms", new OpenCvSharp.Point(10, 70), HersheyFonts.HersheySimplex, 1, Scalar.Red, 2);
  212.                 Cv2.PutText(frame, "postprocessTime:" + yoloV8.postprocessTime.ToString("F2") + "ms", new OpenCvSharp.Point(10, 110), HersheyFonts.HersheySimplex, 1, Scalar.Red, 2);
  213.                 Cv2.PutText(frame, "totalTime:" + yoloV8.totalTime.ToString("F2") + "ms", new OpenCvSharp.Point(10, 150), HersheyFonts.HersheySimplex, 1, Scalar.Red, 2);
  214.                 Cv2.PutText(frame, "video fps:" + videoFps.ToString("F2"), new OpenCvSharp.Point(10, 190), HersheyFonts.HersheySimplex, 1, Scalar.Red, 2);
  215.                 Cv2.PutText(frame, "det fps:" + yoloV8.detFps.ToString("F2"), new OpenCvSharp.Point(10, 230), HersheyFonts.HersheySimplex, 1, Scalar.Red, 2);
  216.                 Cv2.PutText(frame, "progress:" + index.ToString() + "/" + frameCount.ToString(), new OpenCvSharp.Point(10, 270), HersheyFonts.HersheySimplex, 1, Scalar.Red, 2);
  217.  
  218.                 Cv2.ImShow("DetectionResult 按下ESC，退出", frame);
  219.  
  220.                 //for test
  221.                 delay = 1;
  222.                 //delay = (int)(delay - _stopwatch.ElapsedMilliseconds);
  223.                 //if (delay <= 0)
  224.                 //{
  225.                 //    delay = 1;
  226.                 //}
  227.                 //Console.WriteLine("delay:" + delay.ToString()) ;
  228.                 if (Cv2.WaitKey(delay) == 27 || Cv2.GetWindowProperty("DetectionResult 按下ESC，退出", WindowPropertyFlags.Visible) < 1.0)
  229.                 {
  230.                     Cv2.DestroyAllWindows();
  231.                     vcapture.Release();
  232.                     break; // 如果按下ESC，退出循环
  233.                 }
  234.             }
  235.  
  236.             Cv2.DestroyAllWindows();
  237.             vcapture.Release();
  238.         }
  239.  
  240.         /// <summary>
  241.         /// 异步接口-视频推理
  242.         /// </summary>
  243.         /// <param name="sender"></param>
  244.         /// <param name="e"></param>
  245.         private void button1_Click(object sender, EventArgs e)
  246.         {
  247.             if (video_path == "")
  248.             {
  249.                 MessageBox.Show("请先选择视频！");
  250.                 return;
  251.             }
  252.  
  253.             if (output_path == "")
  254.             {
  255.                 MessageBox.Show("请先选择输出目录！");
  256.                 return;
  257.             }
  258.  
  259.             images_path = output_path + "\\output\\images";
  260.             labels_path = output_path + "\\output\\labels";
  261.  
  262.             if (!Directory.Exists(images_path))
  263.             {
  264.                 Directory.CreateDirectory(images_path);
  265.             }
  266.  
  267.             if (!Directory.Exists(labels_path))
  268.             {
  269.                 Directory.CreateDirectory(labels_path);
  270.             }
  271.  
  272.             textBox1.Text = "开始异步推理检测";
  273.  
  274.             Application.DoEvents();
  275.  
  276.             Thread thread = new Thread(new ThreadStart(VideoDetectionAsync));
  277.  
  278.             thread.Start();
  279.             thread.Join();
  280.  
  281.             textBox1.Text = "异步推理检测完成！";
  282.         }
  283.  
  284.         void VideoDetectionAsync()
  285.         {
  286.             vcapture = new VideoCapture(video_path);
  287.             if (!vcapture.IsOpened())
  288.             {
  289.                 MessageBox.Show("打开视频文件失败");
  290.                 return;
  291.             }
  292.  
  293.             Mat frame = new Mat();
  294.             List<DetectionResult> detResults;
  295.  
  296.             // 获取视频的fps
  297.             double videoFps = vcapture.Get(VideoCaptureProperties.Fps);
  298.             // 计算等待时间（毫秒）
  299.             int delay = (int)(1000 / videoFps);
  300.             Stopwatch _stopwatch = new Stopwatch();
  301.  
  302.             int frameCount = (int)vcapture.Get(VideoCaptureProperties.FrameCount);
  303.  
  304.             Cv2.NamedWindow("DetectionResult 按下ESC，退出", WindowFlags.Normal);
  305.             Cv2.ResizeWindow("DetectionResult 按下ESC，退出", vcapture.FrameWidth / 2, vcapture.FrameHeight / 2);
  306.  
  307.             int FrameWidth = vcapture.FrameWidth;
  308.             int FrameHeight = vcapture.FrameHeight;
  309.             int index = 0;
  310.  
  311.             while (vcapture.Read(frame))
  312.             {
  313.                 if (frame.Empty())
  314.                 {
  315.                     MessageBox.Show("读取失败");
  316.                     return;
  317.                 }
  318.  
  319.                 //输出图片
  320.                 Task.Factory.StartNew(() =>
  321.                 {
  322.                     Cv2.ImWrite(images_path + "\\" + video_name + "_" + index.ToString() + ".jpg", frame);
  323.                 });
  324.  
  325.                 detResults = yoloV8Async.Detect(frame);
  326.  
  327.                 StringBuilder sbLabels = new StringBuilder();
  328.                 //绘制结果
  329.                 foreach (DetectionResult r in detResults)
  330.                 {
  331.                     Cv2.PutText(frame, $"{r.Class}:{r.Confidence:P0}", new OpenCvSharp.Point(r.Rect.TopLeft.X, r.Rect.TopLeft.Y - 10), HersheyFonts.HersheySimplex, 1, Scalar.Red, 2);
  332.                     Cv2.Rectangle(frame, r.Rect, Scalar.Red, thickness: 2);
  333.  
  334.                     //class, x_center, y_center, width, height
  335.                     double width = Math.Round(r.Rect.Width / (double)FrameWidth, 6);
  336.                     double height = Math.Round(r.Rect.Height / (double)FrameHeight, 6);
  337.                     double x_center = Math.Round((r.Rect.Right - (r.Rect.Width / 2)) / (double)FrameWidth, 6);
  338.                     double y_center = Math.Round((r.Rect.Bottom - (r.Rect.Height / 2)) / (double)FrameHeight, 6);
  339.  
  340.                     sbLabels.Append($"{r.ClassId} {x_center} {y_center} {width} {height}\n");
  341.                 }
  342.  
  343.                 //生成不带BOM格式
  344.                 using (FileStream fs = new FileStream(labels_path + "\\" + video_name + "_" + index.ToString() + ".txt", FileMode.Create))
  345.                 {
  346.                     using (StreamWriter sw = new StreamWriter(fs, new UTF8Encoding(false)))
  347.                     {
  348.                         sw.Write(sbLabels.ToString());
  349.                     }
  350.                 }
  351.                 //Cv2.ImWrite(images_path + "\\" + index.ToString() + ".jpg", frame);
  352.                 index++;
  353.  
  354.                 Cv2.PutText(frame, "preprocessTime:" + yoloV8Async.preprocessTime.ToString("F2") + "ms", new OpenCvSharp.Point(10, 30), HersheyFonts.HersheySimplex, 1, Scalar.Red, 2);
  355.                 Cv2.PutText(frame, "inferTime:" + yoloV8Async.inferTime.ToString("F2") + "ms", new OpenCvSharp.Point(10, 70), HersheyFonts.HersheySimplex, 1, Scalar.Red, 2);
  356.                 Cv2.PutText(frame, "postprocessTime:" + yoloV8Async.postprocessTime.ToString("F2") + "ms", new OpenCvSharp.Point(10, 110), HersheyFonts.HersheySimplex, 1, Scalar.Red, 2);
  357.                 Cv2.PutText(frame, "totalTime:" + yoloV8Async.totalTime.ToString("F2") + "ms", new OpenCvSharp.Point(10, 150), HersheyFonts.HersheySimplex, 1, Scalar.Red, 2);
  358.                 Cv2.PutText(frame, "video fps:" + videoFps.ToString("F2"), new OpenCvSharp.Point(10, 190), HersheyFonts.HersheySimplex, 1, Scalar.Red, 2);
  359.                 Cv2.PutText(frame, "det fps:" + yoloV8Async.detFps.ToString("F2"), new OpenCvSharp.Point(10, 230), HersheyFonts.HersheySimplex, 1, Scalar.Red, 2);
  360.                 Cv2.PutText(frame, "progress:" + index.ToString() + "/" + frameCount.ToString(), new OpenCvSharp.Point(10, 270), HersheyFonts.HersheySimplex, 1, Scalar.Red, 2);
  361.  
  362.                 //Cv2.ImWrite(images_path + "\\" + index.ToString() + ".jpg", frame);
  363.  
  364.                 Cv2.ImShow("DetectionResult 按下ESC，退出", frame);
  365.  
  366.                 // for test
  367.                 delay = 1;
  368.                 //delay = (int)(delay - _stopwatch.ElapsedMilliseconds);
  369.                 //if (delay <= 0)
  370.                 //{
  371.                 //    delay = 1;
  372.                 //}
  373.                 //Console.WriteLine("delay:" + delay.ToString()) ;
  374.                 if (Cv2.WaitKey(delay) == 27 || Cv2.GetWindowProperty("DetectionResult 按下ESC，退出", WindowPropertyFlags.Visible) < 1.0)
  375.                 {
  376.                     Cv2.DestroyAllWindows();
  377.                     vcapture.Release();
  378.                     break; // 如果按下ESC，退出循环
  379.                 }
  380.             }
  381.             Cv2.DestroyAllWindows();
  382.             vcapture.Release();
  383.         }
  384.     }
  385. }

下载 

源码下载

天天代码码天天
微信公众号
.NET 人工智能实践
注：本文转载自blog.csdn.net的天天代码码天天的文章"https://lw112190.blog.csdn.net/article/details/140008554"。版权归原作者所有，此博客不拥有其著作权，亦不承担相应法律责任。如有侵权，请联系我们删除。
复制链接
相关推荐
发表评论

评论记录：

未查询到任何数据！