/data_generator_object_detection_2d

A data generator for 2D object detection

Primary LanguagePythonGNU General Public License v3.0GPL-3.0

Data Generator for 2D Object Detection


img01 img01
img01 img01

Contents

  1. Overview
  2. Dependencies
  3. How to use it

Overview

This is a data generator for 2D object detection with axis-aligned rectangular bounding boxes. Its main focus is on flexible and easily extensible data augmentation capabilities.

Main features:

  • Lets you easily define arbitrary chains of image transformations.
  • Provides most common photometric and geometric image transformations out of the box.
  • Provides both random and deterministic versions of all image transformations.
  • Provides flexible options for how to decide whether a random transformation of an image is "valid" (see tutorial).
  • Provides parsers for ground truth annotations in the MS COCO (JSON) and Pascal VOC (XML) formats as well as for a wide range of CSV formats.

Dependencies

  • Python 3.x
  • Numpy
  • OpenCV
  • Beautiful Soup 4.x

Python 2 compatibility: This implementation seems to work with Python 2.7, but I don't provide any support for it. It's 2018 and nobody should be using Python 2 anymore.

How to use it

Everything you need to know in order to use this data generator is explained in the following Jupyter notebook tutorial: