dataset convertor
V 0.0.2 Release
PASCAL VOC : el conjunto de datos PASCAL VOC tiene un archivo XML para cada imagen.
YOLO : YOLO DataSet tiene un archivo txt para cada imagen.
COCO : Coco DataSet tiene un archivo JSON para cada imagen.
Actualmente, los siguientes formatos son compatibles:
| de | a | implementado |
|---|---|---|
| VOCO PASCAL | Yolo (archivos txt) | Sí |
| Yolo | Pascal VOC (archivos XML) | Sí |
| de | a | Problema/PR (si corresponde) |
|---|---|---|
| VOCO PASCAL | Coco (archivos json) | No |
| VOCO PASCAL | TFrecord (archivos TFRecord) | No |
| PALMA DE COCO | Pascal VOC (archivos XML) | No |
| PALMA DE COCO | Yolo (archivos txt) | No |
| PALMA DE COCO | TFrecord (archivos TFRecord) | No |
| Yolo | Coco (archivos json) | No |
| Yolo | TFrecord (archivos TFRecord) | No |
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-convertorConvierta anotaciones de un formato a otro formato.
Ejemplo de formato del conjunto de datos:
- 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 ()Cree un problema/PR Si falta algún formato. La contribución de la fuente abierta es bienvenida. Compruebe la guía que contribuye para obtener más detalles.
