| 123456789101112131415161718 |
- from ultralytics import YOLO
- if __name__ == "__main__":
- model = YOLO('yolov12n.yaml')
- model.load("yolov12n.pt")
- # Train the model
- results = model.train(
- data='/Users/jsonxu/Desktop/yolov12/dataset/data.yaml',
- epochs=100,
- batch=16,
- imgsz=640,
- scale=0.5, # S:0.9; M:0.9; L:0.9; X:0.9
- mosaic=1.0,
- mixup=0.0, # S:0.05; M:0.15; L:0.15; X:0.2
- copy_paste=0.1, # S:0.15; M:0.4; L:0.5; X:0.6
- )
|
