数据集准备

现代目标检测、实例分割和旋转框检测

选择包含 data.yaml 的数据集根目录;旋转框页面也允许输入 YAML 路径。YAML 中指定训练集、验证集和类别映射。图片与标签须一一对应,类别编号从 0 开始且与类别表一致。

dataset/
  data.yaml
  images/train/
  images/val/
  labels/train/
  labels/val/
任务 每个目标一行的标签内容
普通检测 类别编号 中心x 中心y 宽 高
实例分割 类别编号 x1 y1 x2 y2 …,目标轮廓多边形
旋转框检测 类别编号 x1 y1 x2 y2 x3 y3 x4 y4,四个角点

坐标归一化至 0~1。旋转框角点按顺时针或逆时针排列;普通五列框、原始 DOTA 标签及自相交四边形不能直接用作该页面的 OBB 标签。软件会预检查 OBB 标签并报告文件及行号。

现代分类

按类别组织训练/验证图片,类别名称使用对应子文件夹名称。推荐显式保留 train/类别名 与 val/类别名,由分类引擎按其数据规则读取。此页面不是经典分类 TXT 清单导入入口。

classification/
  train/OK/...
  train/NG/...
  val/OK/...
  val/NG/...

分类结果目录生成在数据集同级的“数据集名_runs”,避免训练结果被误识别成类别。不要把以前生成的结果目录混进类别目录。

经典目标检测:一个 COCO 文件夹

默认选择 COCO 模式,选择数据集根目录并点击“检查数据集”。软件识别训练图片、COCO JSON、类别及可选的独立验证集。常见结构如下:

coco/
  images/train/...
  images/val/...             # 可选
  annotations/train.json
  annotations/val.json      # 可选

没有独立验证集时,全部训练图片用于训练,不自动划分,也不计算 mAP。OK 背景图片允许没有目标框;保留图片记录,annotations 中不必为它伪造框,也不需要新建一个“OK 目标”类别。软件会保留这些负样本。

类别映射以训练标注为基准;训练与验证类别名称须一致。软件生成实验内的标注副本与类别清单,不修改原始 COCO 文件。

经典分类:一个分类文件夹

优先读取 train_list.txt、val_list.txt 和 Label.txt;没有清单时识别 train/类别名、val/类别名。保留已有训练/验证划分,不自动重新划分。类别编号须连续,验证类别不能超出训练类别。

如果已经有其他命名结构,先使用“检查数据集”,以软件识别的训练数量、验证数量及类别为准。不要通过手动伪造类别编号来跳过检查。

异常检测:OK 图片与独立校准

训练目录只放 OK 图片。可以选择独立 OK 校准目录;留空时按文件内容去重后留出约 20% 做校准,至少需要 3 张不同 OK 图。训练特征库不使用校准图片;内容重复的训练图和校准图会被拒绝。

这条自动留出规则仅适用于异常检测,不能套用到经典 COCO 检测。