from ultralytics import YOLO

# 加载训练好的模型
model = YOLO('run/train/weights/best.pt')

# 在验证集上进行推理
results = model.predict(
    source='datasets/valid/images/-10-1_mp4-1_jpg.rf.fb00c51bb30dded64f20f66f13047b9d.jpg',  # 替换为您的验证集图像路径或单张图片路径
    imgsz=640,
    conf=0.25,
    iou=0.45,
    save=True,
    save_txt=True,
    save_conf=True,
    project='run',
    name='train'
)