16 lines
427 B
Python
16 lines
427 B
Python
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')
|
|
|
|
|
|
|