通过.Net平台使用Yolov8

Yolov8简介

YOLOv8是Ultralytics公司推出的基于对象检测模型的YOLO最新系列,它能够提供截至目前最先进的对象检测性能。

借助于以前的YOLO模型版本支持技术,YOLOv8模型运行得更快、更准确,同时为执行任务的训练模型提供了统一的框架,这包括:

  • 目标检测

  • 实例分割

  • 图像分类

YOLOv8也非常高效和灵活,它可以支持多种导出格式,而且该模型可以在CPU和GPU上运行。

使用Yolov8.Net

从GitHub上Clone Yolov8.Net。项目结构如下:

通过.Net平台使用Yolov8_第1张图片

配置训练环境

Yolov8支持CPU和GPU进行推理。首先需要安装CUDA Toolkit和cuDNN模块,具体安装方法见教程,按照教程安装即可。CUDA与cuDNN安装教程(超详细)。

测试cuDNN是否正确配置

  1. 打开CUDA安装目录:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\extras\demo_suite

  1. 右键此处打开PowerShell

  1. 执行bandwidthTest.exe,结尾出现PASS

通过.Net平台使用Yolov8_第2张图片
  1. 执行deviceQuery.exe,结尾出现PASS

通过.Net平台使用Yolov8_第3张图片

完成单元测试

通过.Net平台使用Yolov8_第4张图片

使用异常

  1. [ErrorCode:RuntimeException] FAIL : LoadLibrary failed with error 126

这个异常很常见,在Yolov5Net、Yolov7Net项目中issue中都能找到这个异常。主要是在使用GPU推理,即参数useGPU = True时导致异常。

异常详情:

"H:\Yolov8.Net\test\Yolov8net.test\bin\Debug\net6.0\runtimes\win-x64\native\onnxruntime_providers_cuda.dll"
 Microsoft.ML.OnnxRuntime.OnnxRuntimeException : [ErrorCode:RuntimeException] FAIL : LoadLibrary failed with error 126 "" when trying to load "\runtimes\win-x64\native\onnxruntime_providers_cuda.dll"

异常原因:

cuDNN模块未配置。安装【配置训练环境】中的教程依次安装CUDN和配置cuDNN。

你可能感兴趣的:(深度学习,pytorch)