Skip to content

V 0.0.2 Release

Latest
Compare
Choose a tag to compare
@codeperfectplus codeperfectplus released this 17 Jun 13:23
· 3 commits to main since this release

Current support format

Currently, the following formats are supported:

from to implemented
PASCAL VOC YOLO(TXT files) Yes
YOLO PASCAL VOC (XML files) Yes

Upcoming support format

from to Issue/PR(if any)
PASCAL VOC COCO (JSON files) No
PASCAL VOC TFRecord (TFRecord files) No
COCO PASCAL VOC (XML files) No
COCO YOLO (TXT files) No
COCO TFRecord (TFRecord files) No
YOLO COCO (JSON files) No
YOLO TFRecord (TFRecord files) No

Installation

Installation from source code

git clone https://github.com/codePerfectPlus/dataset-convertor/
cd dataset-convertor
python -m venv venv
source venv/bin/activate
pip install requirements.txt

Installation from PyPI

pip install dataset-convertor

Usage

convert annotations from one format to another format.

dataset formatting example:

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

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

Pascal VOC(xml) to yolo(txt)

from convert import Convertor

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

from yolo(txt) to Pascal VOC(xml)

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