如何在YOLO的model.predict()中指定文件保存路径及自定义文件名？

一、更改保存路径 #

Ultralytics YOLO提供两种直接修改保存路径的方式：

• 方式1：使用project和name参数
  指定根目录和子目录名，预测结果会自动保存到project/name路径下：

  from ultralytics import YOLO
  model = YOLO('best.pt')
  results = model.predict(
      source='/Users/sereentaleb/Desktop/download.jpg',
      save=True,
      conf=0.25,
      project='/Users/sereentaleb/Desktop/YOLO_检测结果',  # 自定义根文件夹
      name='自定义子文件夹'  # 自定义子文件夹名
  )
  

  执行后文件会保存到/Users/sereentaleb/Desktop/YOLO_检测结果/自定义子文件夹/download.jpg。

• 方式2：直接用save_dir指定完整目录
  不需要分层文件夹的话，直接指定最终保存目录即可：

  from ultralytics import YOLO
  model = YOLO('best.pt')
  results = model.predict(
      source='/Users/sereentaleb/Desktop/download.jpg',
      save=True,
      conf=0.25,
      save_dir='/Users/sereentaleb/Desktop/直接保存的文件夹'
  )
  

  文件会保存到指定目录下，保留原文件名。

二、自定义检测后的文件名 #

YOLO默认沿用源文件的文件名，要自定义名称需在预测完成后手动重命名，通过results对象获取保存路径后执行操作：

from ultralytics import YOLO
import os

model = YOLO('best.pt')
results = model.predict(
    source='/Users/sereentaleb/Desktop/download.jpg',
    save=True,
    conf=0.25,
    save_dir='/Users/sereentaleb/Desktop/YOLO_检测结果'
)

# 获取已保存的文件路径
saved_file = os.path.join(results[0].save_dir, os.path.basename(results[0].path))
# 自定义新文件名
new_file_name = '我的检测结果.jpg'
new_file_path = os.path.join(results[0].save_dir, new_file_name)
# 执行重命名
os.rename(saved_file, new_file_path)

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