如何仅检测yolov8中的人员类别？

Question

我只想检测 yolov8 中的人员类别，还有一个人可以告诉我怎么做吗？

我在文档中找不到任何内容。我只想裁剪第一人称并将其放入分类模型中。

Answer 1

以下代码可以实现您想要的功能。

pip install ultralytics

$ python3
>>> from ultralytics import YOLO
>>> model = YOLO("yolov8n.pt")
Downloading https://github.com/ultralytics/assets/releases/download/v0.0.0/yolov8n.pt to yolov8n.pt...
100%|██████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 6.23M/6.23M [00:09<00:00, 685kB/s]
>>> results = model("zidane.jpg")

image 1/1 /home/.../zidane.jpg: 384x640 2 persons, 1 tie, 7.9ms
Speed: 0.7ms preprocess, 7.9ms inference, 0.9ms postprocess per image at shape (1, 3, 640, 640)
>>> results = model("zidane.jpg", classes=0)

image 1/1 /home/.../zidane.jpg: 384x640 2 persons, 5.7ms
Speed: 0.3ms preprocess, 5.7ms inference, 2.5ms postprocess per image at shape (1, 3, 640, 640) 

只需添加参数

classes=

，如文档中here所述。

您将从推理输出中获得边界框，并可以使用它来裁剪图像。

Answer 2

我不知道如何才能只跟踪一个人，但只检测人：

from ultralytics import YOLO
from ultralytics.utils.plotting import Annotator

model = YOLO('yolov8n.pt')
model.classes = [0]

这对我有用。