YOLO预训练模型中的检测类别

COCO数据集

YOLO模型（如YOLOv8）通常在COCO数据集上预训练，其中类别0代表“person”（行人）。可以加载预训练模型后，仅保留类别为0的检测结果，在官网上也有相关的解释，就 COCO 数据集而言，相关与训练模型的文件被保存在 coco.yaml 文件中

COCO -Ultralytics YOLO 文档

ultralytics/ultralytics/cfg/datasets/coco.yaml at main · ultralytics/ultralytics · GitHub.

因此，如果我们想要使用预训练模型单独输出某一个或某几个类别，通过对照COCO数据集中的yaml文件就可以找到相关类别的序号是多少

运行程序

官网提供了YOLO的相关参数

Predict - Ultralytics YOLO Docs