yolov8系列[三]-训练自己的模型
方法1
- 增加配置文件
yolo copy-cfg
会copyultralytics/yolo/cfg/default.yaml
文件到根目录default_copy.yaml
yolo copy-cfg
- 修改配置文件
修改模型文件、数据文件、epoch、batch、projects、device等 - 训练模型-支持多卡训练
nohup yolo cfg=default_drone.yaml >logs/drone.log 2>&1 &
1.1 配置文件介绍
- –weights: 指定YOLO模型的权重文件路径。
- –source: 指定输入数据源,可以是图像文件、视频文件、摄像头设备或包含文件路径的文本文件。
- –img-size: 指定图像的输入尺寸,用于调整模型的输入大小。
- –conf-thres: 目标置信度阈值,低于该阈值的目标将被过滤掉。
- –iou-thres: 非极大值抑制(NMS)的IOU(交并比)阈值,用于合并重叠的边界框。
- –device: 指定计算设备,