https://github.com/pierluigiferrari/data_generator_object_detection_2d

A data generator for 2D object detection
https://github.com/pierluigiferrari/data_generator_object_detection_2d

data-augmentation data-generator image-transformations object-detection

Last synced: 10 months ago
JSON representation

A data generator for 2D object detection

• Host: GitHub
• URL: https://github.com/pierluigiferrari/data_generator_object_detection_2d
• Owner: pierluigiferrari
• License: gpl-3.0
• Created: 2018-03-29T22:45:34.000Z (almost 8 years ago)
• Default Branch: master
• Last Pushed: 2018-04-16T12:35:09.000Z (almost 8 years ago)
• Last Synced: 2025-04-02T10:11:09.758Z (11 months ago)
• Topics: data-augmentation, data-generator, image-transformations, object-detection
• Language: Python
• Size: 32.7 MB
• Stars: 85
• Watchers: 2
• Forks: 38
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE.txt

Awesome Lists containing this project

README

          
## Data Generator for 2D Object Detection

---

| | |

|---|---|

| ![img01](examples/image01_original.png) | ![img01](examples/image01_processed01.png) |

| ![img01](examples/image01_processed02.png) | ![img01](examples/image01_processed03.png) |

### Contents

1. [Overview](#overview)

2. [Dependencies](#dependencies)

3. [How to use it](#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:

* [Data Generator Tutorial](data_generator_tutorial.ipynb)