yolov8自带的predict函数相关参数说明
在使用这个函数时,可以通过不同的参数来控制预测过程的行为。: 输入图像的尺寸,可以是单个整数(长宽相等)或元组(长和宽)。: NMS(非极大值抑制)的交并比阈值,用于消除重叠的检测框。: 置信度阈值,滤除低于该置信度的检测结果。: 是否在保存的文本文件中包含置信度。: 是否使用类别无关的 NMS。: 是否保存裁剪后的检测结果。: 是否保存带有检测框的图像。: 指定要检测的类的索引列表。: 保存预测
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)
更多推荐
所有评论(0)