如何在 Colab 中更改 yolov8 模型的训练模型位置

Question

这是在colab中训练模型的命令

!yolo task=detect mode=train model=yolov8s.pt data=coco.yaml epochs=10 imgsz=640 

我想将模型的保存位置从 /runs/exp 更改为 /content/drive/MyDrive/yolov8，我该怎么做

我尝试了 --weights 参数和其他一些参数，但没有成功

Answer 1

您可以使用project=要保存结果的文件夹名称；即，就你而言，project=/content/drive/MyDrive/yolov8.

这样，就会在你的yolov8文件夹中创建一个train文件夹，并将结果保存在其中。如果您想更改名称“train”，您可以提及名称=文件夹名称

您可以在此处阅读有关这些参数的更多信息：https://github.com/ultralytics/ultralytics/blob/main/ultralytics/yolo/cfg/default.yaml