封装caffe-windows-gpu(支持模型有多个输出)

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/sinat_30071459/article/details/53735600

之前封装的代码只能有一个输出:http://blog.csdn.net/sinat_30071459/article/details/51823390

并且有点问题,现在作了修改,模型可以有多个输出(当然也可以只有一个输出),可用于做图片的多属性分类,如:


这个例子有四个输出;

代码输出的分类结果为:

typedef std::pair<int, float> Prediction;
vector<vector<Prediction> > predictions = classifier.Classify(img, 2);
int为标签下标(0,1,2,3...),float为确信度。predictions为模型所有输出的结果,每个结果取确信度排前二的情况。

如果想显示具体的标签,可自行修改代码。

代码下载地址:http://download.csdn.net/detail/sinat_30071459/9715053


猜你喜欢

转载自blog.csdn.net/sinat_30071459/article/details/53735600