我可以使用 detect.py yolov5 仅检测特定标签吗

Question

我已经在自定义数据集上训练了一个基于 YOLOv5 的模型，该数据集有两个 classes（例如人和汽车） 我正在使用 detect.py 和以下命令：

> python detect.py --weights best.pt --source video.mp4

我只想检测汽车class而不检测人，如何实现？

Answer 1

我认为你可以使用 detect.py 的参数 --类。只需使用 类.

的索引

Answer 2

您可以指定要检测的 类[--类] 将使用参数。

例子

python detect.py --weights "your weights.pt" --source "video/image/stream" --classes 0,1,2

在上面的命令中，0,1,2是classId，所以当你运行它时，只会检测到提到的类。