python - 如何使用 TensorFlow 在 ROI 周围创建边界框

Question

我正在使用 inception v3 和 tensorflow 来识别图像中的一些对象。但是，它只是创建了一个可能对象的列表，我需要它来告知它们在图像中的位置。

我正在学习鲜花教程:https://www.tensorflow.org/versions/r0.9/how_tos/image_retraining/index.html

bazel-bin/tensorflow/examples/image_retraining/retrain --image_dir ~/flower_photos

Answer 1

Inception 是一个分类网络，而不是定位网络。

您需要另一种架构来预测边界框，例如 R-CNN及其更新(和更快)的变体(Fast R-CNN、Faster R-CNN)。

可选地，如果您想使用 inception 并且您有一个带有类和边界框坐标注释的训练集，您可以向 inception 添加一个回归头，并使网络学习回归边界框坐标。这与迁移学习相同，但您只需使用最后一个卷积层输出作为特征提取器，并训练这个新头部为训练集中的每个边界框回归 4 个坐标 + 1 个类。