Ultrlytics、YOLO 继续、追加、减少训练

1、继续训练

需要开启 resume = Ture

import warnings

warnings.filterwarnings('ignore')
from ultralytics import RTDETR

# https://docs.ultralytics.com/modes/train/

if __name__ == '__main__':
    model = RTDETR('last.pt path') # 替换你的权重文件
    # model.load('') # loading pretrain weights
    model.train(data='ultralytics/cfg/datasets/VisDrone.yaml',
                cache=False,
                imgsz=640,
                epochs=200,
                batch=8,
                workers=8,
                device='0',
                resume=True,  # last.pt path
                project='runs/train',
                name='exp',
                # amp=True
                )

2、追加、减少训练

1. 1首先在 ultralytics/engine/trainer.py  的 119行左右找到 self.epoch

        self.epochs = self.args.epochs  # 改成你需要的训练次数
        self.start_epoch = 0 # 从哪次开始训练

1. 2训练文件更改（ 不需要开启 resume）

import warnings

warnings.filterwarnings('ignore')
from ultralytics import RTDETR

# https://docs.ultralytics.com/modes/train/

if __name__ == '__main__':
    model = RTDETR('last.pt path') # 替换你的最后权重文件
    # model.load('') # loading pretrain weights
    model.train(data='ultralytics/cfg/datasets/VisDrone.yaml',
                cache=False,
                imgsz=640,
                epochs=200,
                batch=8,
                workers=8,
                device='0',
                # resume=True,  # last.pt path
                project='runs/train',
                name='exp', # 修改为你训练的文件夹
                # amp=True
                )

1. 3训练完后记得改回源码！！！！

        self.epochs = self.args.epochs  # 
        self.start_epoch = 0 #