• Stars
    star
    267
  • Rank 153,621 (Top 4 %)
  • Language
    Python
  • License
    MIT License
  • Created over 5 years ago
  • Updated over 2 years ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

Convert VOC format XMLs to COCO format json

voc2coco

This is script for converting VOC format XMLs to COCO format json(ex. coco_eval.json).

Why we need to convert VOC xmls to COCO format json ?

We can use COCO API, this is very useful(ex. calculating mAP).

How to use

1. Make labels.txt

labels.txt if need for making dictionary for converting label to id.

Sample labels.txt

Label1
Label2
...

In order to get all labels from your *.xml files, you can use this command in shell:

grep -REoh '<name>.*</name>' /Path_to_Folder | sort | uniq

This will search for all name tags in VOC.xml files, then show unique ones. You can also go further and create labels.txt file.

grep -ERoh '<name>(.*)</name>' /Path_to_folder | sort | uniq | sed 's/<name>//g' | sed 's/<\/name>//g' > labels.txt

2. Run script

2.1 Usage 1(Use ids list)
$ python voc2coco.py \
    --ann_dir /path/to/annotation/dir \
    --ann_ids /path/to/annotations/ids/list.txt \
    --labels /path/to/labels.txt \
    --output /path/to/output.json \
    <option> --ext xml
2.2 Usage 2(Use annotation paths list)

Sample paths.txt

/path/to/annotation/file.xml
/path/to/annotation/file2.xml
...
$ python voc2coco.py \
    --ann_paths_list /path/to/annotation/paths.txt \
    --labels /path/to/labels.txt \
    --output /path/to/output.json \
    <option> --ext xml

3. Example of usage

In this case, you can convert Shenggan/BCCD_Dataset: BCCD Dataset is a small-scale dataset for blood cells detection. by this script.

$ python voc2coco.py
    --ann_dir sample/Annotations \
    --ann_ids sample/dataset_ids/test.txt \
    --labels sample/labels.txt \
    --output sample/bccd_test_cocoformat.json \
    --ext xml

# Check output
$ ls sample/ | grep bccd_test_cocoformat.json
bccd_test_cocoformat.json

# Check output
cut -f -4 -d , sample/bccd_test_cocoformat.json
{"images": [{"file_name": "BloodImage_00007.jpg", "height": 480, "width": 640, "id": "BloodImage_00007"}