11 lines
316 B
Python
11 lines
316 B
Python
from ultralytics import YOLO
|
|
|
|
pretender = r"yolov8s.pt"
|
|
dataPath = r"/mnt/d/fiber/yolov8/ultralytics-main/ultralytics-main/fourDataset.yaml"
|
|
# Load a model
|
|
model = YOLO("yolov8s.yaml").load(pretender)
|
|
|
|
|
|
model.train(data=dataPath, epochs=300, device=[0, 1, 2], imgsz=1200, batch=30, patience=0)
|
|
metrics = model.val()
|