https://github.com/Py-Contributors/dataset-convertor
Dataset-convertor tool in python for object detection dataset
https://github.com/Py-Contributors/dataset-convertor
coco-json convert-data-format dataset-converter hacktoberfest object-detection object-detetection pascal-voc pip-package pypi-package python python3 ssd-mobilenet tensorflow tfrecords tfrecords-files yolo yolov5
Last synced: about 12 hours ago
JSON representation
Dataset-convertor tool in python for object detection dataset
- Host: GitHub
- URL: https://github.com/Py-Contributors/dataset-convertor
- Owner: Py-Contributors
- License: mit
- Created: 2021-12-06T16:55:08.000Z (over 4 years ago)
- Default Branch: main
- Last Pushed: 2022-10-18T06:35:29.000Z (over 3 years ago)
- Last Synced: 2026-04-03T01:43:25.265Z (3 months ago)
- Topics: coco-json, convert-data-format, dataset-converter, hacktoberfest, object-detection, object-detetection, pascal-voc, pip-package, pypi-package, python, python3, ssd-mobilenet, tensorflow, tfrecords, tfrecords-files, yolo, yolov5
- Language: Python
- Homepage:
- Size: 19.5 KB
- Stars: 6
- Watchers: 1
- Forks: 6
- Open Issues: 2
-
Metadata Files:
- Readme: README.md
- Contributing: CONTRIBUTING.md
- License: LICENSE
Awesome Lists containing this project
README
👉 Convert object detection dataset format 👈
## Dataset types
`PASCAL VOC`: Pascal voc dataset have a xml file for each image.
`YOLO`: YOLO dataset have a txt file for each image.
`COCO`: COCO dataset have a json file for each image.
## 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
```bash
git clone https://github.com/codePerfectPlus/dataset-convertor/
```
```bash
cd dataset-convertor
python -m venv venv
```
```bash
source venv/bin/activate
pip install requirements.txt
```
### Installation from PyPI
```bash
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)
```bash
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)
```bash
from convert import Convertor
con = Convertor(input_folder='/home/user/data/yolo5', output_folder='/home/user/data/pascal_voc')
con.yolo2voc()
```
## Contributing
create an issue/PR if any format is missing.Open-source contribution is welcome.check the [contributing guide](/CONTRIBUTING.md) for details.
## Reference
- 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
## License
- [MIT License](/LICENSE)
## Authors
- [CodePerfectPlus](https://github/com/CodePerfectPlus)