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



我对图像/视频中的对象检测有一些问题。

我成功地尝试了这个教程:具有6行代码的列车目标检测AI

在本教程中,我们训练一个模型来检测全息透镜。如果我想检测汽车,我需要训练一个新的模型。

我的问题是:我能拿我的实际模型(全息透镜)来检测全息透镜和汽车吗?

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

对不起,我的英语不好!!

是的,您可以在任意数量的对象类上训练模型。本教程基于YOLOV3,它是最先进的对象检测模型之一,速度足够快,可以在实时和视频中使用。在您提到的页面上,在步骤1中,它解释了如何准备和注释数据集。你的数据可以有很多类别的物体(如汽车、全息透镜等),但你需要在训练前设置这个数字。此外,在注释步骤(为对象指定标签)中,您应该为对象指定正确的名称。

此页面上有一个链接,说明如何准备自己的数据:目标检测培训——准备您的自定义数据集

如果数据已经准备好了,请确保它是Pascal Voc格式的,因为有必要与您使用的模型保持一致。最后,您只需要一个模型来检测所有类型的对象。准确度取决于迭代次数、每个类别中的对象数量、对象的形状、大小、颜色、角度等的多样性等等

最新更新