dataset convertor
V 0.0.2 Release
PASCAL VOC : Pascal VOC DataSet a un fichier XML pour chaque image.
YOLO : Yolo Dataset a un fichier txt pour chaque image.
COCO : Coco DataSet a un fichier JSON pour chaque image.
Actuellement, les formats suivants sont pris en charge:
| depuis | à | mis en œuvre |
|---|---|---|
| Pascal COV | Yolo (fichiers txt) | Oui |
| Yolo | Pascal COV (fichiers XML) | Oui |
| depuis | à | Problème / PR (le cas échéant) |
|---|---|---|
| Pascal COV | Coco (fichiers JSON) | Non |
| Pascal COV | Tfrecord (fichiers tfrecord) | Non |
| Coco | Pascal COV (fichiers XML) | Non |
| Coco | Yolo (fichiers txt) | Non |
| Coco | Tfrecord (fichiers tfrecord) | Non |
| Yolo | Coco (fichiers JSON) | Non |
| Yolo | Tfrecord (fichiers tfrecord) | Non |
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-convertorConvertissez les annotations d'un format en un autre format.
Exemple de mise en forme de l'ensemble de données:
- 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 ()Créez un problème / PR si un format est manquant. La contribution de l'Open-source est la bienvenue. Consultez le guide de contribution pour plus de détails.
