如何使用python接口調(diào)用已訓(xùn)練好的caffe模型測試分類-創(chuàng)新互聯(lián)
小編給大家分享一下如何使用python接口調(diào)用已訓(xùn)練好的caffe模型測試分類，相信大部分人都還不怎么了解，因此分享這篇文章給大家參考一下，希望大家閱讀完這篇文章后大有收獲，下面讓我們一起去了解一下吧！
創(chuàng)新互聯(lián)是一家專注于成都網(wǎng)站制作、成都網(wǎng)站設(shè)計、外貿(mào)營銷網(wǎng)站建設(shè)與策劃設(shè)計,彭山網(wǎng)站建設(shè)哪家好?創(chuàng)新互聯(lián)做網(wǎng)站,專注于網(wǎng)站建設(shè)十余年,網(wǎng)設(shè)計領(lǐng)域的專業(yè)建站公司;建站業(yè)務(wù)涵蓋:彭山等地區(qū)。彭山做網(wǎng)站價格咨詢:028-86922220訓(xùn)練好了model后，可以通過python調(diào)用caffe的模型，然后進(jìn)行模型測試的輸出。
本次測試主要依靠的模型是在caffe模型里面自帶訓(xùn)練好的結(jié)構(gòu)參數(shù)：~/caffe/models/bvlc_reference_caffenet/bvlc_reference_caffenet.caffemodel,以及結(jié)構(gòu)參數(shù)
：~/caffe/models/bvlc_reference_caffenet/deploy.prototxt相結(jié)合，用python接口進(jìn)行調(diào)用。
訓(xùn)練的源代碼以及相應(yīng)的注釋如下所示：
# -*- coding: UTF-8 -*-
import os
import caffe
import numpy as np
root='/home/zf/caffe/'#指定根目錄
deploy=root+'models/bvlc_reference_caffenet/deploy.prototxt'#結(jié)構(gòu)文件
caffe_model=root + 'models/bvlc_reference_caffenet/bvlc_reference_caffenet.caffemodel'
#已經(jīng)訓(xùn)練好的model
dir =root+'examples/images/'#保存測試圖片的集合
filelist=[]
filenames=os.listdir(dir)
for fn in filenames:
fullfilename = os.path.join(dir,fn)
filelist.append(fullfilename)
#filelist.append(fn)
def Test(img):
#加載模型
net = caffe.Net(deploy,caffe_model,caffe.TEST)
# 加載輸入和配置預(yù)處理
transformer = caffe.io.Transformer({'data':net.blobs['data'].data.shape})
transformer.set_mean('data', np.load('/home/zf/caffe/python/caffe/imagenet/ilsvrc_2012_mean.npy').mean(1).mean(1))
transformer.set_transpose('data', (2,0,1))
transformer.set_channel_swap('data', (2,1,0))
transformer.set_raw_scale('data', 255.0)
#注意可以調(diào)節(jié)預(yù)處理批次的大小
#由于是處理一張圖片，所以把原來的10張的批次改為1
net.blobs['data'].reshape(1,3,227,227)
#加載圖片到數(shù)據(jù)層
im = caffe.io.load_image(img)
net.blobs['data'].data[...] = transformer.preprocess('data', im)
#前向計算
out = net.forward()
# 其他可能的形式 : out = net.forward_all(data=np.asarray([transformer.preprocess('data', im)]))
#預(yù)測分類
print out['prob'].argmax()
#打印預(yù)測標(biāo)簽
labels = np.loadtxt("/home/zf/caffe/data/ilsvrc12/synset_words.txt", str, delimiter='\t')
top_k = net.blobs['prob'].data[0].flatten().argsort()[-1]
print 'the class is:',labels[top_k]
f=file("/home/zhengfeng/caffe/examples/zf/label.txt","a")
f.writelines(img+' '+labels[top_k]+'\n')
labels_filename=root +'data/ilsvrc12/synset_words.txt'
#循環(huán)遍歷文件夾root+'examples/images/'下的所有圖片
for i in range(0,len(filelist)):
img=filelist[i]
Test(img)ps:主要有以下的文件需要說明
待測試的文件夾里面的圖片數(shù)據(jù)為：
最后的輸出結(jié)果如下：
以下是本人定義的label.txt文件寫入的預(yù)測的數(shù)據(jù)：
如果在編譯的時候出現(xiàn)import caffe error的話，說明沒有導(dǎo)入caffe
Export PYTHONPATH=$PYTHONPATH:/home/zf/caffe/python，如果還是不行，可能是你的caffe的python接口未編譯，cd /home/zf/caffe，然后執(zhí)行make pycaffe,接著再測試。
以上是“如何使用python接口調(diào)用已訓(xùn)練好的caffe模型測試分類”這篇文章的所有內(nèi)容，感謝各位的閱讀！相信大家都有了一定的了解，希望分享的內(nèi)容對大家有所幫助，如果還想學(xué)習(xí)更多知識，歡迎關(guān)注創(chuàng)新互聯(lián)成都網(wǎng)站設(shè)計公司行業(yè)資訊頻道！
另外有需要云服務(wù)器可以了解下創(chuàng)新互聯(lián)scvps.cn，海內(nèi)外云服務(wù)器15元起步，三天無理由+7*72小時售后在線，公司持有idc許可證，提供“云服務(wù)器、裸金屬服務(wù)器、高防服務(wù)器、香港服務(wù)器、美國服務(wù)器、虛擬主機、免備案服務(wù)器”等云主機租用服務(wù)以及企業(yè)上云的綜合解決方案，具有“安全穩(wěn)定、簡單易用、服務(wù)可用性高、性價比高”等特點與優(yōu)勢，專為企業(yè)上云打造定制，能夠滿足用戶豐富、多元化的應(yīng)用場景需求。
分享名稱：如何使用python接口調(diào)用已訓(xùn)練好的caffe模型測試分類-創(chuàng)新互聯(lián)
網(wǎng)頁鏈接：http://m.jiaotiyi.com/article/dhggei.html