dataset convertor

Converter o formato do conjunto de dados de detecção de objetos?

Tipos de dados

PASCAL VOC : o conjunto de dados Pascal VOC possui um arquivo XML para cada imagem.

YOLO : o conjunto de dados YOLO possui um arquivo txt para cada imagem.

COCO : o conjunto de dados Coco possui um arquivo JSON para cada imagem.

Formato de suporte atual

Atualmente, os seguintes formatos são suportados:

de	para	implementado
Pascal voc	YOLO (arquivos txt)	Sim
YOLO	Pascal voc (arquivos XML)	Sim

Próximo formato de suporte

de	para	Problema/PR (se houver)
Pascal voc	Coco (arquivos JSON)	Não
Pascal voc	Tfrecord (arquivos Tfrecord)	Não
COCO	Pascal voc (arquivos XML)	Não
COCO	YOLO (arquivos txt)	Não
COCO	Tfrecord (arquivos Tfrecord)	Não
YOLO	Coco (arquivos JSON)	Não
YOLO	Tfrecord (arquivos Tfrecord)	Não

Instalação

Instalação do código -fonte

git clone https://github.com/codePerfectPlus/dataset-convertor/

 cd dataset-convertor
python -m venv venv

 source venv/bin/activate
pip install requirements.txt

Instalação da Pypi

pip install dataset-convertor

Uso

Converta anotações de um formato em outro formato.

Exemplo de formatação do conjunto de dados:

 - data/pascal_voc/JPEGImages/*.jpg
- data/pascal_voc/Annotations/*.xml

- data/yolo5/JPEGImages/*.jpg
- data/yolo5/labels/*.txt

Pascal Voc (XML) para YOLO (TXT)

from convert import Convertor

con = Convertor(input_folder= ' /home/user/data/pascal_voc ' , output_folder= ' /home/user/data/yolo5 ' )
con.voc2yolo ()

De Yolo (TXT) ao Pascal VOC (XML)

from convert import Convertor
con = Convertor(input_folder= ' /home/user/data/yolo5 ' , output_folder= ' /home/user/data/pascal_voc ' )
con.yolo2voc ()

Contribuindo

Crie um problema/PR se houver algum formato.

Referência

• Pascal voc - http://host.robots.ox.ac.uk/pascal/voc/
• Coco - http://cocodataset.org/
• YOLO9000 - https://arxiv.org/abs/1612.08242
• YOLO4 - https://arxiv.org/abs/2004.10934v1

Licença

• MIT Licença

Autores

• CodePerfectPlus