如何获取YOLOv8预测结果的保存目录路径或编号？

方法一：从预测结果中直接提取路径 #

YOLOv8的predict方法返回的results对象中，已经包含了结果保存的完整路径，直接通过results[0].save_dir即可获取：

from ultralytics import YOLO

model = YOLO("./yolov8n.pt")

results = model.predict(source="../TEST/doggy.jpg", save=True, save_txt=True)
# 获取完整保存路径
save_path = results[0].save_dir
# 打印路径，示例输出：d:\runs\detect\predict4
print(save_path)

如果需要提取路径里的编号（比如示例中的4），可以通过路径拆分实现：

import os
# 提取目录名，比如"predict4"
dir_name = os.path.basename(save_path)
# 剥离前缀得到编号
predict_number = dir_name.lstrip("predict")
print(predict_number)  # 输出：4

方法二：自定义保存路径（跳过自动编号） #

如果不想使用自动生成的predictN格式目录，可以通过project和name参数指定固定路径，这样路径更可控：

from ultralytics import YOLO

model = YOLO("./yolov8n.pt")

results = model.predict(
    source="../TEST/doggy.jpg",
    save=True,
    save_txt=True,
    project="./my_detection_results",  # 自定义根目录
    name="dog_detect"                  # 自定义子目录名，无自动编号
)
# 获取自定义的保存路径
save_path = results[0].save_dir

内容的提问来源于stack exchange，提问作者Hariharan