我们可以使用经过图像分类训练的模型来帮助进行张量流中的物体检测吗?

时间:2019-06-17 11:12:43

标签: image tensorflow machine-learning deep-learning object-detection

我已经使用 Tensorflow-for-poets 构建图像分类模型。但是,我现在想在对象检测模型中使用经过训练的模型。我可以直接导入.pb文件,还是必须重新训练模型?

尝试时出现此错误

  

KeyError:“名称'image_tensor:0'表示一个不存在的张量。图中的操作'image_tensor'不存在。”

1 个答案:

答案 0 :(得分:2)

您不能直接使用图像分类产生的.pb模型来执行对象检测。您将必须获得一个对象检测模型,对其进行训练,然后使用它进行检测。 Tensorflow对象检测模型zoo中有预训练的对象检测模型。

下面的详细答案:

图像分类和对象检测是两个不同但密切相关的任务。实际上,罗斯·吉尔希克(Ross Girshick)在著名论文R-CNN

上也提出了类似的问题
  

ImageNet上CNN分类结果在多大程度上可以归纳为PASCAL VOC挑战赛上的物体检测结果?

这个问题基本上意味着图像分类模型可以用来帮助物体检测,但是还需要更多步骤。因此,您不能仅直接使用分类网络来执行对象检测任务。 (但是您给出的错误有所不同,您可以找到正确的张量名称并修复该错误,但是直接使用分类网络以这种方式进行对象检测只是没有意义的。)

有一个幼稚的解决方案可以将两者结合起来,您可以使用各种大小的滑动窗口穿过图像并进行分类,从而可以执行对象检测。

另一个解决方案已集成。举个例子,Faster R-CNN是一个对象检测网络,它使用VGG作为特征提取器(在原始论文中)。在这里,您可以看到VGG是图像分类网络,并且已在某些图像分类任务中进行了预训练。

enter image description here

image source

相关问题