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

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"), //根据自己的数据集进行配置
        new(2, "cryptomonas"),
        new(3, "limnothrix"),
        new(4, "navicula"),
        new(5, "raphidiopisis"),
        new(6, "staurastrum"),
        new(7, "unknown")
       

    },

    true
);

上面有一个巨坑需要注意:onnx的输出名称不一致也会报错。

查看onnx模型的名字,可以通过netro.app网站进行查看

其他只需要常规配置图片以及模型路径即可。

  • 9
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值