dataset convertor
V 0.0.2 Release
PASCAL VOC : набор данных Pascal VOC имеет XML -файл для каждого изображения.
YOLO : набор данных YOLO имеет файл TXT для каждого изображения.
COCO : Coco Dataset имеет файл JSON для каждого изображения.
В настоящее время поддерживаются следующие форматы:
| от | к | реализовано |
|---|---|---|
| Pascal Voc | Yolo (файлы TXT) | Да |
| Йоло | Pascal VOC (XML -файлы) | Да |
| от | к | Выпуск/PR (если есть) |
|---|---|---|
| Pascal Voc | Коко (файлы json) | Нет |
| Pascal Voc | Tfrecord (файлы tfrecord) | Нет |
| Коко | Pascal VOC (XML -файлы) | Нет |
| Коко | Yolo (файлы TXT) | Нет |
| Коко | Tfrecord (файлы tfrecord) | Нет |
| Йоло | Коко (файлы json) | Нет |
| Йоло | 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 ()Создайте проблему/PR, если какой-либо формат отсутствует. Взнос с источниками приветствуются. Проверьте руководство для получения подробной информации.
