是否可以使用 TensorFlow 进行边界框预测? 我在 github 上找到了 TensorBox,但我正在寻找一种更好的支持或官方方法来解决这个问题。
我需要为我自己的类(class)重新训练模型。
最佳答案
不清楚你的意思。你需要物体检测吗?我从“边界框”假设它。如果是这样,初始网络不能直接适用于您的任务,它们是分类网络。 您应该寻找对象检测模型,例如单次检测器 (SSD) 或您只看一次 (YOLO)。他们经常使用来自分类网络的预训练卷积层,但在其顶部有额外的层。如果你想要 Inception(又名 GoogLeNet),YOLO 就是基于它的。看看这个实现:https://github.com/thtrieu/darkflow或者您可以在 Google 中找到的任何其他内容。
关于python - 使用 tensorflow 和 inception-v3 的边界框,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/38446536/