YOLOv8只检测人（或其他一种或者多种类别）

参考官方文档预测 -Ultralytics YOLOv8 文档

这里用到两个参数。

save_txt

bool

False

将检测结果保存在文本文件中，格式如下 [class] [x_center] [y_center] [width] [height] [confidence].有助于与其他分析工具集成。

classes

list[int]

None

根据一组类别 ID 过滤预测结果。只有属于指定类别的检测结果才会返回。在多类检测任务中，该功能有助于集中检测相关对象。

 先获取图像的检测结果，用save_txt=True

from ultralytics import YOLO

if __name__ == '__main__':
   model =  YOLO('yolov8n.pt')
   source='../3.png'
   model.predict(source,show= True,save_txt=True)

0 0.12485 0.63357 0.249234 0.730482
62 0.711472 0.751161 0.21186 0.216051
62 0.386564 0.730147 0.139936 0.149018
64 0.473246 0.884826 0.0292761 0.0255563
66 0.726949 0.907271 0.165755 0.0436129
66 0.379293 0.860779 0.118892 0.0450625

运行后结果txt如图，每行最前面的数字就是图中识别对象的类别ID

修改代码

model.predict(source,show= True,classes=[0])

将classes里的数字修改到你想保留的物体id即可，这里我只识别人，人的id是0我就写0