Skip to main content

some tools about pascal voc format dataset

Project description

Pascal VOC Tools

This package includes some tools about pascal voc format dataset like read xml, write xml, resize image.

XmlReader

from pascal_voc_tools import XmlReader

>>> xml_path = './test.xml'
>>> reader = XmlReader(xml_path)
>>> ann_dict = reader.load()

som functions for reading a xml file and geting data in it.

XmlWriter

from pascal_voc_tools import XmlWriter

>>> writer = XmlWriter(image_path, image_width, image_height, image_depth, database, segmented)
>>> writer.add_object(name, xmin, ymin, xmax, ymax, pose, truncated, difficult)
>>> writer.save(save_path)

Actually, if you have a dict have the format same as loaded dict from XmlReader, you can simply used like:

>>> writer = XmlWriter()
>>> writer.save(save_path, ann_dict)

DatasetResize

from pascal_voc_tools import DatasetResize

>>> resizer = DatasetResize(root_voc_dir, save_voc_dir)
>>> resizer.resize_dataset_by_min_size(min_size)
>>> resizer.copy_imagesets()  # if the file include

DataSplit

from pascal_voc_tools import DataSplit

>>> spliter = DataSplit(root_dir)
>>> result = spliter.split_by_rate(test_rate)
>>> spliter.save(result)

add Augmenter from https://github.com/Paperspace/DataAugmentationForObjectDetection

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

pascal_voc_tools-0.1.17.tar.gz (86.8 kB view hashes)

Uploaded Source

Built Distribution

pascal_voc_tools-0.1.17-py3-none-any.whl (97.3 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page