tensorflow - 可视化增强的火车图像 [tensorflow object detection api]

Question

可以在 tensorflow 对象检测 api 配置文件中增加图像，例如：

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

如何可视化训练图像以检查增强结果？

感谢您的帮助。

Answer 1

我建议您查看input_test.py文件，特别是test_apply_image_and_box_augmentationclass中的函数DataAugmentationFnTest。您可以在data_augmentation_options那里添加并将您的图像传递给tensor_dict. 为了可视化它，您可以在 sess.run() 之后调用 matplotlib 函数，因为augmented_tensor_dict_out它可以作为输入参数传递。

Answer 2

这是实现问题https://github.com/majrie/visualize_augmentation/blob/master/visualize_augmentation.ipynb中所问内容的代码。

它基于@danyfang 的回答。