YOLOv8是Ultralytics公司推出的基于对象检测模型的YOLO最新系列,它能够提供截至目前最先进的对象检测性能。
借助于以前的YOLO模型版本支持技术,YOLOv8模型运行得更快、更准确,同时为执行任务的训练模型提供了统一的框架,这包括:
目标检测
实例分割
图像分类
YOLOv8也非常高效和灵活,它可以支持多种导出格式,而且该模型可以在CPU和GPU上运行。
从GitHub上Clone Yolov8.Net。项目结构如下:
Yolov8支持CPU和GPU进行推理。首先需要安装CUDA Toolkit和cuDNN模块,具体安装方法见教程,按照教程安装即可。CUDA与cuDNN安装教程(超详细)。
打开CUDA安装目录:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\extras\demo_suite
右键此处打开PowerShell
执行bandwidthTest.exe,结尾出现PASS
执行deviceQuery.exe,结尾出现PASS
这个异常很常见,在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。