Skip to content

Commit

Permalink
Add debug image function and debug augmenter
Browse files Browse the repository at this point in the history
  • Loading branch information
aleju committed Dec 1, 2019
1 parent 7566a90 commit fc411a0
Show file tree
Hide file tree
Showing 4 changed files with 1,429 additions and 0 deletions.
13 changes: 13 additions & 0 deletions changelogs/master/added/20191117_debug_images.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,13 @@
# Generate Debug Images #502

* Added module `imgaug.augmenters.debug`.
* Added function `imgaug.augmenters.debug.draw_debug_image()`. The function
draws an image containing debugging information for a provided set of
images and non-image data (e.g. segmentation maps, bounding boxes)
corresponding to a single batch. The debug image visualizes these
informations (e.g. bounding boxes drawn on images) and offers relevant
information (e.g. actual value ranges of images, labels of bounding
boxes and their counts, etc.).
* Added augmenter `imgaug.augmenters.debug.SaveDebugImageEveryNBatches`.
Augmenter corresponding to `draw_debug_image()`. Saves an image at every
n-th batch into a provided folder.
2 changes: 2 additions & 0 deletions imgaug/augmenters/__init__.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
"""Combination of all augmenters, related classes and related functions."""
# pylint: disable=unused-import
from __future__ import absolute_import
from imgaug.augmenters.arithmetic import *
from imgaug.augmenters.artistic import *
Expand All @@ -7,6 +8,7 @@
from imgaug.augmenters.color import *
from imgaug.augmenters.contrast import *
from imgaug.augmenters.convolutional import *
from imgaug.augmenters.debug import *
from imgaug.augmenters.edges import *
from imgaug.augmenters.flip import *
from imgaug.augmenters.geometric import *
Expand Down

0 comments on commit fc411a0

Please sign in to comment.