Labelme转YOLO格式

帮助将LabelMe标注工具的JSON格式转换为YOLO文本文件格式。 如果你已经使用LabelMe标注了分割数据集，可以轻松使用此工具将其转换为YOLO格式数据集。

参数说明

--json_dir LabelMe JSON文件所在文件夹路径。

--val_size (可选) 验证数据集大小，例如0.2表示20%用于验证，80%用于训练。默认值为0.1。

--json_name (可选) 转换单个LabelMe JSON文件。

--seg (可选) 转换为YOLOv5 v7.0实例分割数据集。

使用方法

1. 转换JSON文件，通过--val_size拆分训练和验证数据集

将所有LabelMe JSON文件放在labelme_json_dir下，然后运行以下Python命令。

python labelme2yolo.py --json_dir /home/username/labelme_json_dir/ --val_size 0.2

脚本将在不同文件夹下生成YOLO格式的数据集标签和图像，例如，

# 当指定`--seg'时，"YOLODataset"将变为"YOLODataset_seg"
/home/username/labelme_json_dir/YOLODataset/labels/train/
/home/username/labelme_json_dir/YOLODataset/labels/val/
/home/username/labelme_json_dir/YOLODataset/images/train/
/home/username/labelme_json_dir/YOLODataset/images/val/

/home/username/labelme_json_dir/YOLODataset/dataset.yaml

2. 转换JSON文件，按文件夹拆分训练和验证数据集

如果你已经为LabelMe自行拆分了训练数据集和验证数据集，请将这些文件夹放在labelme_json_dir下，例如，

/home/username/labelme_json_dir/train/
/home/username/labelme_json_dir/val/

将所有LabelMe JSON文件放在labelme_json_dir下。 脚本将按文件夹读取训练和验证数据集。 运行以下Python命令。

python labelme2yolo.py --json_dir /home/username/labelme_json_dir/

脚本将在不同文件夹下生成YOLO格式的数据集标签和图像，例如，

# 当指定`--seg'时，"YOLODataset"将变为"YOLODataset_seg"
/home/username/labelme_json_dir/YOLODataset/labels/train/
/home/username/labelme_json_dir/YOLODataset/labels/val/
/home/username/labelme_json_dir/YOLODataset/images/train/
/home/username/labelme_json_dir/YOLODataset/images/val/

/home/username/labelme_json_dir/YOLODataset/dataset.yaml

3. 转换单个JSON文件

将LabelMe JSON文件放在labelme_json_dir下，然后运行以下Python命令。

python labelme2yolo.py --json_dir /home/username/labelme_json_dir/ --json_name 2.json

脚本将在labelme_json_dir下生成YOLO格式的文本标签和图像，例如，

/home/username/labelme_json_dir/2.text
/home/username/labelme_json_dir/2.png

仅在Centos 7/Python 3.6环境下测试过。