30 lines
619 B
Python
30 lines
619 B
Python
from ultralytics import YOLO
|
|
|
|
model = YOLO('/home/hx/yolo/ultralytics_yolo11-main/yolo11n.pt')
|
|
|
|
# 冻结 DFL 层，防止权重过大
|
|
for name, param in model.model.named_parameters():
|
|
if "dfl.conv" in name:
|
|
param.requires_grad = False
|
|
|
|
# 训练
|
|
results = model.train(
|
|
data='/home/hx/yolo/ultralytics_yolo11-main/ailaidata.yaml',
|
|
epochs=500,
|
|
imgsz=640,
|
|
batch=16,
|
|
workers=4,
|
|
device='0',
|
|
project='runs/train',
|
|
name='exp_ailai_detect',
|
|
exist_ok=False,
|
|
optimizer='AdamW',
|
|
lr0=0.0001,
|
|
lrf=0.01,
|
|
patience=50,
|
|
amp=True,
|
|
close_mosaic=10,
|
|
val=True,
|
|
)
|
|
|