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

Question

我正在使用初始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

初始是分类网络,而不是本地化网络.

您需要另一种架构来预测边界框,例如R-CNN及其更新(和更快)的变体(快速R-CNN,更快的R-CNN).

(可选)如果要使用初始化并且您有一个使用类和边界框坐标注释的训练集,则可以将回归头添加到初始状态,并使网络学习回归边界框坐标.这与转移学习是一样的,但你只需使用最后一个卷积层输出作为特征提取器,并训练这个新头部为你的训练集中的每个边界框回归4个坐标+ 1个类.