資源簡介
調用python接口使用googlenet進行圖像識別,代碼可以改了路徑直接用~~~~~~~~~~~~
代碼片段和文件信息
#?coding:?utf-8
import?caffe
import?numpy?as?np
import?matplotlib.pyplot?as?plt
import?os
import?PIL
from?PIL?import?Image
import?sys
#定義caffe根目錄
caffe_root=‘E:/graduate_student/deep_learning/caffe/new_Win_caffe/document/1/caffe-windows/caffe-windows/‘
#網絡結構描述文件
deploy_file=caffe_root+‘models/bvlc_googlenet/deploy.prototxt‘
#訓練好的模型
model_file=caffe_root+‘models/bvlc_googlenet/bvlc_googlenet.caffemodel‘
#cpu模式
caffe.set_mode_cpu()
#定義網絡模型
net?=?caffe.Classifier(deploy_file?#調用deploy文件
???????????????????????model_file??#調用模型文件
???????????????????????mean=np.load(caffe_root?+‘python/caffe/imagenet/ilsvrc_2012_mean.npy‘).mean(1).mean(1)?#調用均值文件
???????????????????????channel_swap=(210)??#caffe中圖片是BGR格式,而原始格式是RGB,所以要轉化
???????????????????????raw_scale=255?????????#python中將圖片存儲為[0?1],而caffe中將圖片存儲為[0?255],所以需要一個轉換
???????????????????????image_dims=(224?224))?#輸入模型的圖片要是224*224的圖片?我們不需要對網上下載的圖片做預處理,程序會自動處理成224*224
#分類標簽文件
imagenet_labels_filename?=?caffe_root?+‘models/bvlc_googlenet/synset
- 上一篇:最詳細神經網絡python描寫附注釋
- 下一篇:python實現圖片拼接
評論
共有 條評論