10.模型训练_转换
模型训练_转换
模型训练_转换1、模型训练2、模型转换参考资料
完成数据集标注的教程内容后，我们就可以使用主板开始训练模型。
该教程只介绍CLI的模型训练和转换，自己可以参考官网进行Python案例修改
1、模型训练
使用CLI命令直接训练模型：将yolo11n.pt文件复制到配置文件所在目录，然后在配置文件所在目录打开终端：
xxxxxxxxxx cd /home/jetson/ultralytics/ultralytics/data/yahboom_data/orange_data x yolo detect train data=orange.yaml model=yolo11n.pt epochs=100 imgsz=640
data：数据集配置文件
model：预训练模型文件
epochs：训练的轮数
imgsz：输入指定图像尺寸
2、模型转换
最终的模型会生成在runs文件夹下：一般选择best.pt文件进行使用
xxxxxxxxxx /home/jetson/ultralytics/ultralytics/data/yahboom_data/orange_data/runs/detect/train/weights
将PyTorch格式的模型转化成TensorRT：
xxxxxxxxxx cd /home/jetson/ultralytics/ultralytics/data/yahboom_data/orange_data/runs/detect/train/weights xxxxxxxxxx yolo export model=best.pt format=engine
