105 lines
2.4 KiB
C#
105 lines
2.4 KiB
C#
using OrpaonVision.Core.Results;
|
|
using OrpaonVision.SiteApp.Runtime.Contracts;
|
|
|
|
namespace OrpaonVision.SiteApp.Runtime.Services;
|
|
|
|
/// <summary>
|
|
/// YOLO推理服务接口。
|
|
/// </summary>
|
|
public interface IYoloInferenceService
|
|
{
|
|
/// <summary>
|
|
/// 初始化模型。
|
|
/// </summary>
|
|
Result Initialize();
|
|
|
|
/// <summary>
|
|
/// 推理单张图像。
|
|
/// </summary>
|
|
Result<IReadOnlyList<YoloDetectionResult>> Predict(byte[] imageData, int width, int height, string pixelFormat = "BGR8Packed");
|
|
|
|
/// <summary>
|
|
/// 推理批量图像。
|
|
/// </summary>
|
|
Result<IReadOnlyList<IReadOnlyList<YoloDetectionResult>>> PredictBatch(
|
|
IReadOnlyList<byte[]> imageBatch,
|
|
IReadOnlyList<(int width, int height)> dimensions,
|
|
string pixelFormat = "BGR8Packed");
|
|
|
|
/// <summary>
|
|
/// 获取支持的类别列表。
|
|
/// </summary>
|
|
IReadOnlyList<string> GetSupportedClasses();
|
|
|
|
/// <summary>
|
|
/// 获取模型信息。
|
|
/// </summary>
|
|
YoloModelInfo GetModelInfo();
|
|
|
|
/// <summary>
|
|
/// 是否已初始化。
|
|
/// </summary>
|
|
bool IsInitialized { get; }
|
|
|
|
/// <summary>
|
|
/// 预热模型。
|
|
/// </summary>
|
|
Result Warmup(int warmupCount = 3);
|
|
|
|
/// <summary>
|
|
/// 释放资源。
|
|
/// </summary>
|
|
void Dispose();
|
|
}
|
|
|
|
/// <summary>
|
|
/// YOLO模型信息。
|
|
/// </summary>
|
|
public sealed class YoloModelInfo
|
|
{
|
|
/// <summary>
|
|
/// 模型名称。
|
|
/// </summary>
|
|
public string ModelName { get; set; } = string.Empty;
|
|
|
|
/// <summary>
|
|
/// 模型版本。
|
|
/// </summary>
|
|
public string ModelVersion { get; set; } = string.Empty;
|
|
|
|
/// <summary>
|
|
/// 输入尺寸。
|
|
/// </summary>
|
|
public (int Width, int Height) InputSize { get; set; }
|
|
|
|
/// <summary>
|
|
/// 支持的类别数量。
|
|
/// </summary>
|
|
public int ClassCount { get; set; }
|
|
|
|
/// <summary>
|
|
/// 类别名称列表。
|
|
/// </summary>
|
|
public IReadOnlyList<string> ClassNames { get; set; } = [];
|
|
|
|
/// <summary>
|
|
/// 是否使用GPU。
|
|
/// </summary>
|
|
public bool UseGpu { get; set; }
|
|
|
|
/// <summary>
|
|
/// 推理时间（毫秒）。
|
|
/// </summary>
|
|
public double InferenceTimeMs { get; set; }
|
|
|
|
/// <summary>
|
|
/// 模型文件大小（字节）。
|
|
/// </summary>
|
|
public long ModelFileSize { get; set; }
|
|
|
|
/// <summary>
|
|
/// 加载时间（毫秒）。
|
|
/// </summary>
|
|
public double LoadTimeMs { get; set; }
|
|
}
|