dataset convertor

Konversi format dataset deteksi objek?

Jenis Dataset

PASCAL VOC : Dataset Pascal VOC memiliki file XML untuk setiap gambar.

YOLO : Dataset Yolo memiliki file txt untuk setiap gambar.

COCO : Dataset Coco memiliki file JSON untuk setiap gambar.

Format Dukungan Saat Ini

Saat ini, format berikut didukung:

dari	ke	diterapkan
VOC Pascal	YOLO (file txt)	Ya
Yolo	Pascal VOC (file XML)	Ya

Format Dukungan Mendatang

dari	ke	Masalah/PR (jika ada)
VOC Pascal	Coco (file JSON)	TIDAK
VOC Pascal	Tfrecord (file TFRECORD)	TIDAK
KELAPA	Pascal VOC (file XML)	TIDAK
KELAPA	YOLO (file txt)	TIDAK
KELAPA	Tfrecord (file TFRECORD)	TIDAK
Yolo	Coco (file JSON)	TIDAK
Yolo	Tfrecord (file TFRECORD)	TIDAK

Instalasi

Instalasi dari kode sumber

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

 cd dataset-convertor
python -m venv venv

 source venv/bin/activate
pip install requirements.txt

Instalasi dari PYPI

pip install dataset-convertor

Penggunaan

Konversi anotasi dari satu format ke format lain.

Contoh pemformatan dataset:

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

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

VOC Pascal (XML) ke YOLO (TXT)

from convert import Convertor

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

Dari Yolo (TXT) ke Pascal VOC (XML)

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

Berkontribusi

Buat masalah/PR jika ada format yang hilang. Kontribusi sumber terbuka diterima. Periksa panduan yang berkontribusi untuk detailnya.

Referensi

• 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

Lisensi

• Lisensi MIT

Penulis

• Codeperfectplus