Model.predict()使用参数save_dir更改保存路径

Model.predict()使用参数save_dir更改保存路径

model.predict('data/images/horses.jpg', conf=0.5, show=True, save=True, save_dir='img')

原因：源码，如果默认配置文件default.yaml中没有save_dir的键，会删除所给**kwargs形成的args字典的键。

if "save_dir" not in cfg:
	overrides.pop("save_dir", None)  # special override keys to ignore

导致合并默认cfg和overrides后也没有save_dir的值，会使用默认路径runs/detect。

check_dict_alignment(cfg, overrides)
cfg = {**cfg, **overrides}  # merge cfg and overrides dicts (prefer overrides)

解决
找到venv里的ultralytics/cfg/default.yaml文件
任意处添加

# Save Path
save_dir: None

然后

model.predict('data/images/horses.jpg', conf=0.5, show=True, save=True, save_dir='img')

就保存在img下。
也不知道为什么默认配置没有save_dir的参数。。。