dataset convertor
V 0.0.2 Release
PASCAL VOC :Pascal VOC数据集为每个图像都有一个XML文件。
YOLO :YOLO数据集为每个图像都有一个TXT文件。
COCO :可可数据集为每个图像都有一个JSON文件。
目前,支持以下格式:
| 从 | 到 | 实施的 |
|---|---|---|
| Pascal VOC | Yolo(TXT文件) | 是的 |
| YOLO | Pascal VOC(XML文件) | 是的 |
| 从 | 到 | 发行/PR(如果有) |
|---|---|---|
| Pascal VOC | 可可(JSON文件) | 不 |
| Pascal VOC | tfrecord(tfrecord文件) | 不 |
| 可可 | Pascal VOC(XML文件) | 不 |
| 可可 | Yolo(TXT文件) | 不 |
| 可可 | tfrecord(tfrecord文件) | 不 |
| YOLO | 可可(JSON文件) | 不 |
| YOLO | tfrecord(tfrecord文件) | 不 |
git clone https://github.com/codePerfectPlus/dataset-convertor/ cd dataset-convertor
python -m venv venv source venv/bin/activate
pip install requirements.txtpip install dataset-convertor将注释从一种格式转换为另一种格式。
数据集格式示例:
- data/pascal_voc/JPEGImages/*.jpg
- data/pascal_voc/Annotations/*.xml
- data/yolo5/JPEGImages/*.jpg
- data/yolo5/labels/*.txt
from convert import Convertor
con = Convertor(input_folder= ' /home/user/data/pascal_voc ' , output_folder= ' /home/user/data/yolo5 ' )
con.voc2yolo ()from convert import Convertor
con = Convertor(input_folder= ' /home/user/data/yolo5 ' , output_folder= ' /home/user/data/pascal_voc ' )
con.yolo2voc ()如果缺少任何格式,请创建一个问题/公关。欢迎开放源贡献。请符合详细信息的贡献指南。