版本260406
This commit is contained in:
@@ -0,0 +1,87 @@
|
||||
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";
|
||||
}
|
||||
Reference in New Issue
Block a user