File size: 332 Bytes
aa24fe8 | 1 2 3 4 5 6 7 | 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')#数据集源文件存放处
|