来自rooneysh/Labelme2YOLO

Labelme2YOLO

帮助将 LabelMe Annotation Tool JSON 格式转换为 YOLO 文本文件格式。如果您已经通过 LabelMe 标记了您的分割数据集，则可以轻松使用此工具来帮助转换为 YOLO 格式的数据集。

参数说明

--json_dir LabelMe JSON 文件的文件夹路径。

--val_size （可选）验证数据集大小，例如 0.2 表示 20% 用于验证。

--test_size （可选）测试数据集大小，例如 0.2 表示测试的 20%。

--json_name （可选）转换单个 LabelMe JSON 文件。

如何使用

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

将所有 LabelMe JSON 文件放在labelme_json_dir下，然后运行这个 python 命令。

python labelme2yolo.py --json_dir /home/username/labelme_json_dir/ --val_size 0.15 --test_size 0.15

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

/home/username/labelme_json_dir/YOLODataset/labels/train/
/home/username/labelme_json_dir/YOLODataset/labels/test/
/home/username/labelme_json_dir/YOLODataset/labels/val/
/home/username/labelme_json_dir/YOLODataset/images/train/
/home/username/labelme_json_dir/YOLODataset/images/test/
/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 格式的数据集标签和图片，例如，

/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

安装

pip install labelme2yolo

执照

labelme2yolo根据MIT许可条款分发。