yolov8自带的predict函数相关参数说明

YOLOv8 是一种先进的目标检测模型。predict 函数是 YOLOv8 中用于进行预测的主要接口。在使用这个函数时，可以通过不同的参数来控制预测过程的行为。以下是一些常见的参数说明：

1.source: 输入源，可以是图像文件路径、视频文件路径、摄像头索引等。
2.model: 预训练模型的路径或模型对象。
3.conf: 置信度阈值，滤除低于该置信度的检测结果。
4.iou: NMS（非极大值抑制）的交并比阈值，用于消除重叠的检测框。
5.device: 运行推理的设备，比如 ‘cpu’ 或 ‘cuda’。
6.imgsz: 输入图像的尺寸，可以是单个整数（长宽相等）或元组（长和宽）。
7.half: 是否使用半精度浮点数进行推理，通常用于加速推理过程。
8.save_txt: 是否将预测结果保存为文本文件。
9.save_conf: 是否在保存的文本文件中包含置信度。
10.save_crop: 是否保存裁剪后的检测结果。
11.save_img: 是否保存带有检测框的图像。
12.classes: 指定要检测的类的索引列表。
13.agnostic_nms: 是否使用类别无关的 NMS。
14.augment: 是否使用数据增强。
15.visualize: 是否可视化特征图。
16.project: 保存预测结果的项目目录。
17.name: 保存预测结果的子目录名称。
18.exist_ok: 是否允许项目目录存在。
19.line_thickness: 绘制检测框的线条粗细。
20.hide_labels: 是否隐藏检测框的标签。
21.hide_conf: 是否隐藏检测框的置信度。

以下是一个使用 predict 函数的示例：

from yolov8 import YOLO

# 加载预训练模型
model = YOLO('yolov8.pt')

# 执行预测
results = model.predict(
    source='path/to/image.jpg',  # 输入源，可以是图像路径、视频路径、摄像头索引等
    conf=0.5,                    # 置信度阈值
    iou=0.45,                    # NMS的交并比阈值
    device='cuda',               # 使用GPU
    imgsz=(640, 640),            # 输入图像的尺寸
    half=False,                  # 不使用半精度浮点数
    save_txt=True,               # 保存预测结果为文本文件
    save_conf=True,              # 在保存的文本文件中包含置信度
    save_crop=False,             # 不保存裁剪后的检测结果
    save_img=True,               # 保存带有检测框的图像
    classes=[0, 1, 2],           # 仅检测指定的类
    agnostic_nms=False,          # 不使用类别无关的NMS
    augment=False,               # 不使用数据增强
    visualize=False,             # 不可视化特征图
    project='runs/detect',       # 保存预测结果的项目目录
    name='exp',                  # 保存预测结果的子目录名称
    exist_ok=False,              # 不允许项目目录存在
    line_thickness=3,            # 绘制检测框的线条粗细
    hide_labels=False,           # 显示检测框的标签
    hide_conf=False              # 显示检测框的置信度
)

# 输出预测结果
print(results)