8 lines
253 B
Python
8 lines
253 B
Python
from ultralytics import YOLO
|
|
|
|
model = YOLO('yolov8n.yaml').load('yolov8n.pt') # build from YAML and transfer weights
|
|
|
|
if __name__ == '__main__':
|
|
model.train(data='ultralytics/datasets/original-license-plates.yaml', epochs=10, imgsz=640, device=0)
|
|
|