File size: 1,061 Bytes
ab39d95 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 |
from ultralytics import YOLO
config = {
# project
'name': 'v11x-p2_coco_pretrained',
'model': 'yolo11x-p2.yaml',
'pretrained': None,
# hyperparams
'lr0': 0.01,
'lrf': 0.01,
'momentum': 0.937,
'weight_decay': 0.0005,
'epochs': 600,
'imgsz': 640,
'batch': 48,
'cos_lr': False,
'close_mosaic': 10,
'warmup_epochs': 3.0,
'warmup_momentum': 0.8,
'warmup_bias_lr': 0.0,
'box': 7.5,
'cls': 0.5,
'dfl': 1.5,
'hsv_h': 0.015,
'hsv_s': 0.7,
'hsv_v': 0.4,
'translate': 0.1,
'scale': 0.9, # coco=0.9
'fliplr': 0.5,
'optimizer': 'SGD',
'patience': 100,
# augment: https://github.com/ultralytics/ultralytics/issues/16607
'mixup': 0.15,
'mosaic': 1.0,
'copy_paste': 0.3,
# system
'seed': 0,
'deterministic': True,
'data': 'ultralytics/cfg/datasets/coco.yaml',
'cache': 'disk',
'device': [0, 1, 2, 3],
'workers': 16,
'project': 'runs/train',
'val': True,
}
model = YOLO(config['model'])
model.train(**config) |