首页
/ YoloDotNet 项目教程

YoloDotNet 项目教程

2024-09-19 12:14:55作者:郦嵘贵Just

1. 项目目录结构及介绍

YoloDotNet 项目的目录结构如下:

YoloDotNet/
├── ConsoleDemo/
│   ├── Program.cs
│   └── ...
├── YoloDotNet/
│   ├── Models/
│   ├── Extensions/
│   ├── Enums/
│   ├── Yolo.cs
│   └── ...
├── test/
│   └── ...
├── .gitattributes
├── .gitignore
├── LICENSE.txt
├── README.md
└── YoloDotNet.sln

目录结构介绍

  • ConsoleDemo/: 包含项目的控制台演示代码,Program.cs 是启动文件。
  • YoloDotNet/: 核心代码目录,包含模型、扩展、枚举等子目录和核心类 Yolo.cs
  • test/: 包含项目的测试代码。
  • .gitattributes: Git 属性配置文件。
  • .gitignore: Git 忽略配置文件。
  • LICENSE.txt: 项目许可证文件。
  • README.md: 项目说明文档。
  • YoloDotNet.sln: 项目解决方案文件。

2. 项目启动文件介绍

项目的启动文件位于 ConsoleDemo/Program.cs。该文件包含了项目的入口点,用于启动和运行 YoloDotNet 的控制台应用程序。

Program.cs 文件内容概述

using System;
using YoloDotNet;

namespace ConsoleDemo
{
    class Program
    {
        static void Main(string[] args)
        {
            // 初始化 Yolo 对象
            var yolo = new Yolo(@"path\to\model.onnx");

            // 加载图像
            var image = Image.Load<Rgba32>(@"path\to\image.jpg");

            // 运行对象检测
            var results = yolo.RunObjectDetection(image, confidence: 0.25, iou: 0.7);

            // 处理结果
            image.Draw(results);
            image.Save(@"path\to\save\image.jpg");
        }
    }
}

启动文件功能

  • 初始化 Yolo 对象: 加载 ONNX 模型。
  • 加载图像: 使用 SixLabors.ImageSharp 加载图像。
  • 运行对象检测: 调用 Yolo 对象的 RunObjectDetection 方法进行对象检测。
  • 处理结果: 在图像上绘制检测结果并保存。

3. 项目配置文件介绍

YoloDotNet 项目没有传统的配置文件(如 .config.yaml 文件),但可以通过代码中的配置选项来调整项目的行为。

配置选项示例

var yolo = new Yolo(new YoloOptions
{
    OnnxModel = @"path\to\model.onnx",
    ModelType = ModelType.ObjectDetection,
    Cuda = true,
    GpuId = 0,
    PrimeGpu = false
});

配置选项说明

  • OnnxModel: 指定 ONNX 模型的路径。
  • ModelType: 指定模型类型,如 ObjectDetection
  • Cuda: 是否启用 CUDA 加速。
  • GpuId: 指定使用的 GPU ID。
  • PrimeGpu: 是否预分配 GPU 内存。

通过这些配置选项,可以在代码中灵活地调整 YoloDotNet 的行为,以适应不同的应用场景。

登录后查看全文
热门项目推荐