arm/yolov5_onnx/Mod/YoloModel.cs
2025-02-04 20:09:10 +08:00

29 lines
1.3 KiB
C#
Raw Permalink Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
namespace yolov5_onnx.Mod
{
public abstract class YoloModel
{
public abstract int Width { get; set; } //輸入影像的寬度
public abstract int Height { get; set; } //輸入影像的高度
public abstract int Depth { get; set; } //輸入影像的深度(通常為3表示RGB影像)
public abstract int Dimensions { get; set; } //數據維度(通常為3對應於寬度、高度和深度)
public abstract int[] Strides { get; set; } //模型的部幅數組
public abstract int[][][] Anchors { get; set; } // 锚點數組,用於目標回歸
public abstract int[] Shapes { get; set; } //模型輸出的形狀數組
public abstract float Confidence { get; set; } //預測信心度閾值
public abstract float MulConfidence { get; set; } //用於多類別預測的信心度縮放因子
public abstract float Overlap { get; set; } //重疊閾值,用於非級大值抑制(NMS)
public abstract string[] Outputs { get; set; } //模型的輸出層名稱數組
public abstract List<YoloLabel> Labels { get; set; } //目標類別的標籤列表
public abstract bool UseDetect { get; set; } //是否使用對象檢測功能
}
}