我是 YOLOv8 的新手,我只想让模型仅检测某些类,而不是模型训练的所有 80 个类。如何指定 YOLOv8 模型仅检测一类?例如仅人。
from ultralytics import YOLO
model = YOLO('YOLOv8m.pt')
Run Code Online (Sandbox Code Playgroud)
我记得我们可以用 YOLOv5 做到这一点,但我不能用 YOLOv8 做到同样的事情:
model = torch.hub.load("ultralytics/yolov5", 'custom', path='yolov5s.pt')
model.classes = [0] # Only person
model.conf = 0.6
Run Code Online (Sandbox Code Playgroud)