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

2024-05-20 10:25:59 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一些关于图像/视频中物体检测的问题。在

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

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

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

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

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


Tags: 对象模型图像视频objectwithtrain教程
1条回答
网友
1楼 · 发布于 2024-05-20 10:25:59

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

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

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

相关问题 更多 >