| from ultralytics import RTDETR | |
| import torch | |
| if __name__ == '__main__': | |
| # 模型加载 | |
| model=RTDETR("/home/lab/LJ/wampee/ultralytics/ultralytics/cfg/models/rt-detr/rtdetr-l.yaml")#.yaml配置文件 | |
| model.train(data="/home/lab/LJ/wampee/ultralytics/train.yaml",epochs=400,batch=4,device = '3')#数据集源文件存放处 | |