9 lines
222 B
Python
9 lines
222 B
Python
from ultralytics import YOLO
|
|
import torch
|
|
|
|
device = torch.device('cuda')
|
|
# device = 'cpu'
|
|
|
|
model = YOLO('yolo11s.pt')
|
|
model.train(data='./yolov11-pecan/data.yaml',device=device,batch=20,epochs=25,workers=16, imgsz = 1280)
|