Визуализация дополненных изображений поездов [API для обнаружения тензорного объекта]

Question

Возможно увеличение изображений в конфигурационных файлах API обнаружения объектов tenorflow, например:

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

Как можно визуализировать тренировочные образы для проверки результатов дополнения?

Спасибо за вашу помощь.

Answer 1

Я предлагаю вам посмотреть файл input_test.py , в частности, функцию test_apply_image_and_box_augmentation в классе DataAugmentationFnTest. Вы можете добавить туда data_augmentation_options и передать свое изображение в tensor_dict. Чтобы визуализировать это, вы можете вызвать функцию matplotlib после sess.run (), так как augmented_tensor_dict_out может быть передано в качестве входного аргумента.

Answer 2

Вот код для выполнения того, что было задано в вопросе https://github.com/majrie/visualize_augmentation/blob/master/visualize_augmentation.ipynb.

Он основан на ответе @ danyfang.