使用官方脚本命令训练时的断点继续训练方法。
训练时的命令(总共训练100epoch):
yolo detect train data=./ultralytics/cfg/datasets/coco.yaml model=./ultralytics/cfg/models/v8/yolov8.yaml project=./run/yolov8/ epochs=100
情况1:还未训练完100epoch,希望继续训练。
方法:将model替换为训练中途的last.pt文件,并且添加resume=True。
detect train data=./ultralytics/cfg/datasets/coco.yaml model=./run/yolov8/train/weights/last.pt project=./run/yolov8/ epochs=100 resume=True
结果:模型将会从断点处开始训练直到100epoch结束。
情况2:已经训练完100epoch,希望增加训练epoch(增加500epoch)。
方法:将epochs替换为500,并且将已有的权重作为pretrained进行加载。
yolo detect train data=./ultralytics/cfg/datasets/coco.yaml model=./ultralytics/cfg/models/v8/yolov8.yaml project=./run/yolov8/ epochs&#