0

我目前正在将 DenseNet_121 dnn 模型用于 ImageNet 数据集,并在 OpenCv 中使用 Caffe 框架对图像进行分类。

我的问题是我只想知道特定对象(例如老虎)是否在框架中,而不是 ImageNet 数据集的所有 1000 个对象的概率。

为了解决这个问题,我创建了一个单独的 my_objects.txt 文件,其中包含我感兴趣的对象。然后我检查 ImageNet 数据集中最有可能出现在框架中的对象是否在该 my_objects.txt 文件中。

问题在于它非常慢(在树莓派上运行)。我的猜测是,这是因为该模型正在计算所有 1000 个对象的概率,而不是仅计算我真正感兴趣的少数对象。

有没有办法从模型中删除其他对象以加快速度?

我拥有的文件是:

  • DenseNet_121.caffemodel
  • DenseNet_121.protobuf
  • classification_classes_ILSVRC2012.txt(列出 ImageNet 数据集的 1000 个对象)
  • my_objects.txt(列出感兴趣的classification_classes_ILSVRC2012的对象)

如果有人可以帮助我,将不胜感激!

4

0 回答 0