ML.Net对象检测和边界框

时间:2020-04-08 11:59:17

标签: deep-learning computer-vision object-detection ml.net

我已经使用Microsoft模型构建器创建了图像分类模型。现在,我需要使用该模型来检测视频流中的对象,并在检测到对象后绘制边界框。我找不到使用从模型构建器生成的模型的c#示例。所有对象检测样本均使用ONNX模型。我尚未找到将为模型生成器生成的model.zip转换为model.onnx的工具。

任何帮助将不胜感激。

1 个答案:

答案 0 :(得分:0)

模型构建器中的图像分类无法检测图像中的对象-这是一个不同的问题。

您可以做的是将图像中对象检测的ONNX样本与您自己的自定义模型结合起来。

基本上,您会运行onnx示例,直到解析边界框为止。然后,通过图像分类器运行图像的该部分,并使用该标签。

这有点骇人听闻,您将很难获得接近实时性能的任何地方。

用于ONNX检测的ONNX示例: https://github.com/dotnet/machinelearning-samples/tree/master/samples/csharp/getting-started/DeepLearning_ObjectDetection_Onnx