之前封裝的代碼只能有一個(gè)輸出：http://blog.csdn.net/sinat_30071459/article/details/51823390

并且有點(diǎn)問(wèn)題，現(xiàn)在作了修改，模型可以有多個(gè)輸出（當(dāng)然也可以只有一個(gè)輸出），可用于做圖片的多屬性分類(lèi)，如：

這個(gè)例子有四個(gè)輸出；

代碼輸出的分類(lèi)結(jié)果為：

[plain]?view plaincopy

typedef?std::pair<int,?float>?Prediction;??

vector<vector<Prediction>?>?predictions?=?classifier.Classify(img,?2);??

int為標(biāo)簽下標(biāo)（0，1，2，3...），float為確信度。predictions為模型所有輸出的結(jié)果，每個(gè)結(jié)果取確信度排前二的情況。

如果想顯示具體的標(biāo)簽，可自行修改代碼。

代碼下載地址：http://download.csdn.net/detail/sinat_30071459/9715053

總結(jié)

以上是生活随笔為你收集整理的封装caffe-windows-gpu(支持模型有多个输出)的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。