如何在Colab中修改YOLOv8训练模型的保存路径?
解决YOLOv8自定义训练结果保存路径的问题
你需要调整的是训练结果的保存路径,而非权重文件路径,因此--weights参数不适用。YOLOv8提供了专门的参数来指定自定义保存路径,以下是两种可行的方法:
方法一:使用project和name参数
通过project指定根目录,name指定该目录下的子文件夹名称,最终训练结果会保存到project/name路径下:
!yolo task=detect mode=train model=yolov8s.pt data=coco.yaml epochs=10 imgsz=640 project=/content/drive/MyDrive/yolov8 name=train_exp
执行后,所有训练日志、权重、结果文件都会保存到/content/drive/MyDrive/yolov8/train_exp目录中。
方法二:直接用save_dir指定完整路径
如果不需要额外的子文件夹,可以直接用save_dir参数指定完整的目标路径:
!yolo task=detect mode=train model=yolov8s.pt data=coco.yaml epochs=10 imgsz=640 save_dir=/content/drive/MyDrive/yolov8
此命令会将所有训练结果直接保存到/content/drive/MyDrive/yolov8目录下,不会自动创建子文件夹。
注意事项
执行前需确保已经将Google Drive挂载到Colab环境中,若未挂载,先执行以下代码完成挂载:
from google.colab import drive drive.mount('/content/drive')
内容的提问来源于stack exchange,提问作者Abhishek Bhadola
相关产品推荐
相关产品推荐

