可视化增强的火车图像[tensorflow对象检测API]

Question

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

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

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

谢谢您的帮助。

Answer 1

我建议您查看input_test.py文件，特别是类DataAugmentationFnTest中的函数test_apply_image_and_box_augmentation。您可以在此处添加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的答案。