7 lines
197 B
Python
7 lines
197 B
Python
from ultralytics import YOLO
|
|
|
|
# Load a pretrained YOLO11 segment model
|
|
model = YOLO("yolo11n-seg.pt")
|
|
|
|
# Train the model
|
|
results = model.train(data="./my_dataset.yaml", epochs=100, imgsz=640) |