2025-10-21 14:11:52 +08:00
|
|
|
from ultralytics import YOLO
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
|
#model = YOLO(r'/home/hx/yolo/ultralytics_yolo11-main/ultralytics/cfg/models/11/yolo11-obb.yaml')
|
2025-12-11 08:37:09 +08:00
|
|
|
model = YOLO(r'/home/hx/yolo/ultralytics_yolo11-main/runs/train/exp_ailai2/weights/best.pt')
|
2025-10-21 14:11:52 +08:00
|
|
|
results = model.train(
|
|
|
|
|
data='data_ailai.yaml',
|
2025-12-11 08:37:09 +08:00
|
|
|
epochs=1000,
|
2025-10-21 14:11:52 +08:00
|
|
|
imgsz=640,
|
|
|
|
|
batch=4,
|
|
|
|
|
workers=10,
|
|
|
|
|
device='0',
|
|
|
|
|
project='runs/train',
|
|
|
|
|
name='exp_ailai',
|
|
|
|
|
exist_ok=False,
|
|
|
|
|
optimizer='AdamW',
|
|
|
|
|
lr0=0.0001,
|
|
|
|
|
patience=0,
|
|
|
|
|
)
|
