有没有办法将经过训练的 Caffe 模型限制为一组类？

Question

我正在使用默认的“bvlc_reference_caffenet”模型。我正在尝试检测抹刀。现在我得到的结果非常令人满意。 spatula 类总是在预测的前 5 类中，但其余的都是无用的随机事物，我永远不会去寻找。我可以在最后添加一个过滤器来删除不需要的结果，但是 Caffe 是否自己提供了这个功能？找不到说的类？

Answer 1

是的，确实如此。 'bvlc_reference_caffenet' 带有一个定义神经网络结构的文本文件。它由输入层、隐藏层集和输出层组成。如果您想让它成为最好的铲子查找器，那么您必须修改输出层并使其成为“铲子”和“世界其他地方”。

请注意，这需要重新训练模型。事实上，如果您只是通过获取现有模型的权重并仅通过用于生成模型的部分迭代来改进模型就足够了。它在计算上仍然会非常昂贵。此外，隐藏层的架构可能不是最优的。

我猜你自己的过滤正是你所需要的。