Search code examples
tensorflowtensorboardobject-detection-apidata-augmentation

Visualizing augmented train images [tensorflow object detection api]


It is possible to augment images in tensorflow object detection api config files, e.g.:

data_augmentation_options {
  random_horizontal_flip {
  }
}
data_augmentation_options {
  ssd_random_crop {
  }
}

How can I visualize the training images to inspect the results of the augmentation?

Thank you for your help.


Solution

  • Here is the code to achieve what has been asked in the question https://github.com/majrie/visualize_augmentation/blob/master/visualize_augmentation.ipynb .

    It is based on the answer of @danyfang.