我可以仅使用dentect.py yolov5检测特定标签
我已经在具有两个类别的自定义数据集上培训了一个基于Yolov5的模型（例如人类和汽车） 我正在使用以下命令使用dentect.py：
> python detect.py --weights best.pt --source video.mp4
我希望仅检测到不检测人类的汽车类，该如何完成？
I have trained a model based on YOLOv5 on a custom dataset which has two classes (for example human and car)
I am using detect.py with the following command:
> python detect.py --weights best.pt --source video.mp4
I want only car class to be detected without detecting humans, how it could be done?
如果你对这篇内容有疑问，欢迎到本站社区发帖提问 参与讨论，获取更多帮助，或者扫码二维码加入 Web 技术交流群。
绑定邮箱获取回复消息
由于您还没有绑定你的真实邮箱，如果其他用户或者作者回复了您的评论，将不能在第一时间通知您！
发布评论
评论（2）
您可以指定要检测 [ - 类] 参数的类。
示例
在上述命令中，0,1,2是分类的，因此，当您运行它时，仅检测到上述类。
You can specify classes, which you want to detect [--classes] arguments will be used.
Example
In above command, 0,1,2 are classId, so when you will run it, only mentioned classes will be detect.
我认为您可以使用该参数 - classe of destect.py。只需使用类的索引即可。
I think you can use the argument --classes of detect.py. Just use the index of the classes.