当前位置:   article > 正文

[C#]winform部署官方yolov8-obb旋转框检测的onnx模型_yolov8 obb 转onnx

yolov8 obb 转onnx

【官方框架地址】

https://github.com/ultralytics/ultralytics
【算法介绍】

Yolov8-obb(You Only Look Once version 8 with Oriented Bounding Boxes)是一种先进的对象检测算法,它在传统的Yolov3和Yolov4基础上进行了优化,加入了OBB(Oriented Bounding Box)旋转框检测,能够更精确地检测并定位出目标物体的位置。

在传统的目标检测算法中,通常使用的是固定方向的边界框(Bounding Box),它假设所有物体在图像中的位置都是相对于图像中心来计算的。然而,这种假设并不总是成立,尤其在处理具有明显方向特征的物体时,固定方向的边界框就无法准确地定位物体的真实位置。

Yolov8-obb通过引入OBB旋转框检测,解决了这一问题。它允许边界框以任意角度存在,更能适应不同方向的目标物体。此外,Yolov8-obb还采用了一种称为“anchor”的机制,通过预设一系列不同大小和方向的锚点框,来逼近真实的物体位置。这种机制不仅提高了检测精度,还大大减少了需要训练的参数数量,提高了算法的效率。

除了旋转框检测,Yolov8-obb还在骨干网络、特征金字塔网络、分类器和回归器等方面进行了优化。例如,它采用了CSPDarknet53作为骨干网络,增强了特征提取能力;采用了多尺度特征融合策略,提高了对不同尺度目标的检测能力;采用了新的非极大值抑制算法,进一步筛选出最有可能的物体位置。

总的来说,Yolov8-obb通过引入旋转框检测和一系列优化策略,提高了目标检测的精度和效率,为计算机视觉领域带来了新的突破。

【效果展示】

【实现部分代码】

  1. using System;
  2. using System.Collections.Generic;
  3. using System.ComponentModel;
  4. using System.Data;
  5. using System.Diagnostics;
  6. using System.Drawing;
  7. using System.Linq;
  8. using System.Text;
  9. using System.Threading.Tasks;
  10. using System.Windows.Forms;
  11. using OpenCvSharp;
  12. namespace FIRC
  13. {
  14. public partial class Form1 : Form
  15. {
  16. Bitmap src = null;
  17. Yolov8ObbManager detector = null;
  18. public Form1()
  19. {
  20. InitializeComponent();
  21. }
  22. private void button1_Click(object sender, EventArgs e)
  23. {
  24. OpenFileDialog openFileDialog = new OpenFileDialog();
  25. openFileDialog.Filter = "图文件(*.*)|*.jpg;*.png;*.jpeg;*.bmp";
  26. openFileDialog.RestoreDirectory = true;
  27. openFileDialog.Multiselect = false;
  28. if (openFileDialog.ShowDialog() == DialogResult.OK)
  29. {
  30. src = new Bitmap(openFileDialog.FileName);
  31. pictureBox1.Image = src;
  32. }
  33. }
  34. private void button2_Click(object sender, EventArgs e)
  35. {
  36. if(pictureBox1.Image==null)
  37. {
  38. return;
  39. }
  40. var result = detector.Inference(src);
  41. var resultImg = detector.DrawImage(src, result);
  42. pictureBox2.Image = resultImg;
  43. }
  44. private void Form1_Load(object sender, EventArgs e)
  45. {
  46. detector = new Yolov8ObbManager(Application.StartupPath+"\\weights\\yolov8s-obb.onnx", Application.StartupPath + "\\weights\\labels.txt");
  47. }
  48. private void button3_Click(object sender, EventArgs e)
  49. {
  50. VideoCapture capture = new VideoCapture(0);
  51. if (!capture.IsOpened())
  52. {
  53. Console.WriteLine("video not open!");
  54. return;
  55. }
  56. Mat frame = new Mat();
  57. var sw = new Stopwatch();
  58. int fps = 0;
  59. while (true)
  60. {
  61. capture.Read(frame);
  62. if (frame.Empty())
  63. {
  64. Console.WriteLine("data is empty!");
  65. break;
  66. }
  67. sw.Start();
  68. var bmp = OpenCvSharp.Extensions.BitmapConverter.ToBitmap(frame);
  69. var result = detector.Inference(bmp);
  70. var resultImg = detector.DrawImage(bmp, result);
  71. sw.Stop();
  72. fps = Convert.ToInt32(1 / sw.Elapsed.TotalSeconds);
  73. sw.Reset();
  74. frame = OpenCvSharp.Extensions.BitmapConverter.ToMat(new Bitmap(resultImg));
  75. Cv2.PutText(frame, "FPS=" + fps, new OpenCvSharp.Point(30, 30), HersheyFonts.HersheyComplex, 1.0, new Scalar(255, 0, 0), 3);
  76. //显示结果
  77. Cv2.ImShow("Result", frame);
  78. int key = Cv2.WaitKey(10);
  79. if (key == 27)
  80. break;
  81. }
  82. capture.Release();
  83. }
  84. }
  85. }


【视频演示】

https://www.bilibili.com/video/BV1ki4y1i7up/?vd_source=989ae2b903ea1b5acebbe2c4c4a635ee
【测试环境】

vs2019,netframework4.7.2,onnxruntime1.16.3
————————————————

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/不正经/article/detail/668993
推荐阅读
相关标签
  

闽ICP备14008679号