11 lines
158 B
Python
11 lines
158 B
Python
|
from ultralytics import YOLO
|
||
|
|
|
||
|
|
model = YOLO("yolo11n.pt")
|
||
|
|
|
||
|
|
model.train(
|
||
|
|
data="data.yaml",
|
||
|
|
epochs=100,
|
||
|
|
imgsz=640,
|
||
|
|
batch=16,
|
||
|
|
name="train1"
|
||
|
|
)
|