training for JC deployment

This commit is contained in:
2025-05-14 16:32:37 +00:00
parent 6011c5005a
commit 787f5a5f23
26 changed files with 141 additions and 0 deletions

8
yolov11-pecan/train.py Normal file
View File

@@ -0,0 +1,8 @@
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)