培训DeepBelief网络以识别多个类别?

时间:2015-12-03 10:32:59

标签: machine-learning computer-vision neural-network caffe object-recognition

DeepBelief框架的学习示例演示了如何训练神经网络识别一个对象类别。用于培训jpcnn_train()的方法没有类别标签参数。

但是,在DeepBelief简单示例中,给定的神经网络可以对多个对象类别进行分类。有没有办法通过DeepBelief进行这种培训?或者我应该查看Caffe并使用它,因为DeepBelief是基于Caffe的?

1 个答案:

答案 0 :(得分:0)

根据他们的文档,特别是函数jpcnn_trainjpcnn_predict的文档,它似乎不支持开箱即用的自定义标签的多类分类。它似乎支持ImageNet标签的多类分类。

但是,您可以训练多个预测变量(here's how to train one),每个自定义类一个,然后选择相应预测变量输出最高值的类。