c#加载yolov5需要注意的事项

最新推荐文章于 2025-09-05 14:25:48 发布

原创

最新推荐文章于 2025-09-05 14:25:48 发布 · 912 阅读

11 ·

CC 4.0 BY-SA版权

文章标签：

#YOLO #c# #计算机视觉

本文介绍了如何在C#中加载Yolov5的初始版本，涉及数据集配置，特别是YoloCocoP5Model中的参数调整，包括识别类别和ONNX模型输出名称的匹配。特别强调了ONNX输出名称一致性的重要性。

1：在c#中加载一个yolov5的初始版本如上图所示。具体的下载连接如下： https://gitcode.com/mirrors/mentalstack/yolov5-net?utm_sourc。

2：配置自己训练的数据集时，需要在yolococoP5Model.cs或者yolococoP6Model.cs文件中修改对应的一些参数。具体如下图所示：

using Yolov5Net.Scorer.Models.Abstract;

namespace Yolov5Net.Scorer.Models;

public record YoloCocoP5Model() : YoloModel(
    640,
    640,
    3,

    12,  //这里的数据=识别个数+5

    new[] { 8, 16, 32 },

    new[]
    {
        new[] { new[] { 010, 13 }, new[] { 016, 030 }, new[] { 033, 023 } },
        new[] { new[] { 030, 61 }, new[] { 062, 045 }, new[] { 059, 119 } },
        new[] { new[] { 116, 90 }, new[] { 156, 198 }, new[] { 373, 326 } }
    },

    new[] { 80, 40, 20 },

    0.20f,
    0.25f,
    0.45f,

    new[] { "output0" },     //注意onnx模型的输出名称要进行查看，确保一致。否则会出现no matching element

    new()
    {
        new(1, "aphanocapsa"), /