29 lines
1.3 KiB
C#
29 lines
1.3 KiB
C#
|
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; } //是否使用對象檢測功能
|
|||
|
|
}
|
|||
|
|
}
|