88 lines
2.2 KiB
C#
88 lines
2.2 KiB
C#
namespace OrpaonVision.SiteApp.Runtime.Contracts;
|
||
|
||
/// <summary>
|
||
/// YOLO推理配置选项。
|
||
/// </summary>
|
||
public sealed class YoloInferenceOptions
|
||
{
|
||
/// <summary>
|
||
/// 运行模式（Real=真实YOLO推理，Mock=模拟模式）。
|
||
/// </summary>
|
||
public string RunMode { get; set; } = "Mock"; // Real | Mock
|
||
|
||
/// <summary>
|
||
/// 模型文件路径。
|
||
/// </summary>
|
||
public string ModelPath { get; set; } = string.Empty;
|
||
|
||
/// <summary>
|
||
/// 标签文件路径。
|
||
/// </summary>
|
||
public string LabelsPath { get; set; } = string.Empty;
|
||
|
||
/// <summary>
|
||
/// 置信度阈值。
|
||
/// </summary>
|
||
public float ConfidenceThreshold { get; set; } = 0.5f;
|
||
|
||
/// <summary>
|
||
/// NMS阈值。
|
||
/// </summary>
|
||
public float NmsThreshold { get; set; } = 0.45f;
|
||
|
||
/// <summary>
|
||
/// 输入图像宽度。
|
||
/// </summary>
|
||
public int InputWidth { get; set; } = 640;
|
||
|
||
/// <summary>
|
||
/// 输入图像高度。
|
||
/// </summary>
|
||
public int InputHeight { get; set; } = 640;
|
||
|
||
/// <summary>
|
||
/// 是否使用GPU。
|
||
/// </summary>
|
||
public bool UseGpu { get; set; } = false;
|
||
|
||
/// <summary>
|
||
/// GPU设备ID。
|
||
/// </summary>
|
||
public int GpuDeviceId { get; set; } = 0;
|
||
|
||
/// <summary>
|
||
/// 批处理大小。
|
||
/// </summary>
|
||
public int BatchSize { get; set; } = 1;
|
||
|
||
/// <summary>
|
||
/// 模型类型（ONNX、TorchScript等）。
|
||
/// </summary>
|
||
public string ModelType { get; set; } = "ONNX";
|
||
|
||
/// <summary>
|
||
/// 是否启用自动预热。
|
||
/// </summary>
|
||
public bool EnableWarmup { get; set; } = true;
|
||
|
||
/// <summary>
|
||
/// 预热次数。
|
||
/// </summary>
|
||
public int WarmupCount { get; set; } = 3;
|
||
|
||
/// <summary>
|
||
/// YOLO推理引擎类型（YoloDotNet、OpenCV、TensorRT等）。
|
||
/// </summary>
|
||
public string InferenceEngine { get; set; } = "YoloDotNet";
|
||
|
||
/// <summary>
|
||
/// ONNX Runtime执行提供程序（CPU、CUDA等）。
|
||
/// </summary>
|
||
public string OnnxRuntimeProvider { get; set; } = "CPU";
|
||
|
||
/// <summary>
|
||
/// 模型版本信息。
|
||
/// </summary>
|
||
public string ModelVersion { get; set; } = "v1.0";
|
||
}
|