跳到主要内容

10.模型训练_转换

模型训练_转换​

模型训练_转换1、模型训练2、模型转换

完成数据集标注的教程内容后，我们就可以使用主板开始训练模型。

​ x 该教程只介绍CLI的模型训练和转换，自己可以参考官网进行Python案例修改

1、模型训练​

使用CLI命令直接训练模型：将yolo26n.pt文件复制到配置文件所在目录，然后在配置文件所在目录打开终端：

xxxxxxxxxx cd /home/jetson/ultralytics/ultralytics/data/yahboom_data/orange_data/ xxxxxxxxxx yolo detect train data=orange.yaml model=/home/jetson/ultralytics/yolo26n.pt epochs=100 imgsz=640

data：数据集配置文件

model：预训练模型文件

epochs：训练的轮数

imgsz：输入指定图像尺寸

image-20260311174549425

image-20260321210208656

2、模型转换​

最终的模型会生成在runs文件夹下：一般选择best.pt文件进行使用

xxxxxxxxxx /home/jetson/ultralytics/ultralytics/data/yahboom_data/orange_data/runs/detect/train2/weights

将pt模型转化成engine模型：

xxxxxxxxxx cd /home/jetson/ultralytics/ultralytics/data/yahboom_data/orange_data/runs/detect/train2/weights xxxxxxxxxx yolo export model=best.pt format=engine

image-20260321210937865