from ultralytics import YOLO


# 加载预训练模型
model = YOLO("yolo11n.pt")
# Use the model
if __name__ == '__main__':
    # Use the model
    results = model.train(data='datasets/TomatoData/data.yaml', epochs=20, batch=4)  # 训练模型
    # results = model.train(data='datasets/TomatoData/data.yaml', epochs=250, batch=4)  # 训练模型
    # 将模型转为onnx格式
    # success = model.export(format='onnx')