我对图像/视频中的物体检测有一些疑问。
我成功地尝试了本教程: Train Object Detection AI with 6 lines of code
在本教程中,我们训练了一个模型来检测Hololens。如果我想检测汽车,就需要训练一种新模型。
我的问题是:我可以采用我的实际模型(hololens)使其检测到hololens和汽车吗?
最后,我想拥有一个文件(模型)来检测我训练过的所有对象。我将放置一个'n'图片的目录,该模型将尝试检测经过训练的所有内容以检测并返回检测到的对象的名称。
对不起,我的英语不好!
答案 0 :(得分:1)
是的,您可以在任意数量的对象类上训练模型。本教程基于 YOLOV3 ,它是最先进的对象检测模型之一,其速度足够快,可以在实时视频中使用。在您提到的页面上,步骤1 中介绍了如何准备和注释数据集。您的数据可以包含许多类别的对象(例如汽车,hololens等),但是您需要在训练之前设置此数字。另外,在注释步骤(为对象添加标签)中,应为对象分配正确的名称。
此页面上有一个链接,该链接说明了如何准备自己的数据: Object Detection Training — Preparing your custom dataset
如果已经准备好数据,请确保其为Pascal Voc格式,因为有必要与使用的模型保持一致。最后,您只需要一种模型即可检测各种对象。精度取决于迭代次数,每个类中的对象数量,对象的形状,大小,颜色,角度等的变化等。