yolo --- 模型训练指导

文件存放路径

数据集标注软件

pip install labelimg

安装好后启动 labelimg

usage: labelImg [-h] [image_dir] [class_file] [save_dir]

(yolov11) PS D:\yolo11\ultralytics> labelimg D:\yolo11\ultralytics\datasets\cs2\images\train\ D:\yolo11\ultralytics\datasets\cs2\labels\train\classes.txt  D:\yolo11\ultralytics\datasets\cs2\labels\train
　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　需要标注的图片路径                                    标签类别                                                    标注文件保存路径

Open Dir：打开需要标注的图片路径

Change Save Dir：标注文件保存路径

View--->Auto Save mode：把这个勾上就不会每到下一张询问你一次

 

按w可以显示画框的坐标十字

按D键可以切换到下一张

注意标注的标签对应的索引需要和数据集配置文件列举的标签索引一样

 

YOLO格式标签文件内容

  • 标签的索引
  • 中心点x坐标 ÷ 图像宽度
  • 中心点y坐标 ÷ 图像高度
  • 矩形框宽度 ÷ 图像宽度
  • 矩形框高度 ÷ 图像高度

标签文件所在文件夹除了生成标签文件，还会标签类别文件classes.txt

数据集配置文件

model.train 需要传入一个数据集文件data，里面指明训练和验证的图片路径、标签类别等

模型配置文件

 

posted @ 2024-11-28 23:51  流水灯  阅读(21)  评论(0)  编辑  收藏  举报