小编TAN*_*TAN的帖子

指定要在预训练模型上使用 YOLOv8 进行检测的类

我是 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)

python detection computer-vision yolo

4
推荐指数
1
解决办法
6097
查看次数

标签 统计

computer-vision ×1

detection ×1

python ×1

yolo ×1