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 ()如果缺少任何格式,請創建一個問題/公關。歡迎開放源貢獻。請符合詳細信息的貢獻指南。