创建自定义检测模型-图像对象检测

问题描述 投票:0回答:1

我对图像/视频中的物体检测有一些疑问。

我成功尝试了本教程:Train Object Detection AI with 6 lines of code

在本教程中,我们训练了一种检测Hololens的模型。如果要检测汽车,则需要训练新模型。

我的问题是:我可以采用我的实际模型(hololens)并使其检测到hololens和汽车吗?

最后,我想拥有一个文件(模型)来检测我放置的所有对象。我将放置'n'个图像的目录,并且该模型将尝试检测经过训练的所有内容以检测并返回检测到的对象的名称。

抱歉,我的英语不好!

python tensorflow object-detection yolo
1个回答
1
投票

是的,您可以在任意数量的对象类上训练模型。本教程基于YOLOV3,它是最先进的对象检测模型之一,其速度足够快,可以在实时视频中使用。在您提到的页面的步骤1中,它说明了如何准备和注释数据集。您的数据可以包含许多类别的对象(例如汽车,hololens等),但是您需要在训练之前设置此数字。另外,在注释步骤(为对象添加标签)中,应为对象分配正确的名称。

此页面上有一个链接,该链接说明了如何准备自己的数据:Object Detection Training — Preparing your custom dataset

如果已经准备好数据,请确保其为Pascal Voc格式,因为有必要与使用的模型保持一致。最后,您只需要一种模型即可检测各种对象。精度取决于迭代次数,每个类别中的对象数量,对象的形状,大小,颜色,角度等的变化等。

© www.soinside.com 2019 - 2024. All rights reserved.