dataset convertor
V 0.0.2 Release
PASCAL VOC : Pascal VOC -Datensatz verfügt über eine XML -Datei für jedes Bild.
YOLO : YOLO -Datensatz hat eine TXT -Datei für jedes Bild.
COCO : Coco -Datensatz hat eine JSON -Datei für jedes Bild.
Derzeit werden die folgenden Formate unterstützt:
| aus | Zu | implementiert |
|---|---|---|
| Pascal VOC | Yolo (txtdateien) | Ja |
| Yolo | Pascal VOC (XML -Dateien) | Ja |
| aus | Zu | Ausgabe/PR (falls vorhanden) |
|---|---|---|
| Pascal VOC | Coco (JSON -Dateien) | NEIN |
| Pascal VOC | TFRECORD (TFRECORD -Dateien) | NEIN |
| Coco | Pascal VOC (XML -Dateien) | NEIN |
| Coco | Yolo (txtdateien) | NEIN |
| Coco | TFRECORD (TFRECORD -Dateien) | NEIN |
| Yolo | Coco (JSON -Dateien) | NEIN |
| Yolo | TFRECORD (TFRECORD -Dateien) | NEIN |
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-convertorKonvertieren Sie Annotationen von einem Format in ein anderes Format.
Beispiel für Datensatzformatierung:
- 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 ()Erstellen Sie ein Problem/PR, wenn ein Format fehlt. Open-Source-Beitrag ist willkommen. Überprüfen Sie den Beitrag zum beitragenden Leitfaden für Einzelheiten.
