python caffe 训练自己的模型_使用训练好的caffe模型分类图片(python版)

英文官方文档：http://nbviewer.jupyter.org/github/BVLC/caffe/blob/master/examples/00-classification.ipynb

导入python caffe包

import numpy as np

import matplotlib.pyplot as plt

# display plots in this notebook

%matplotlib inline

# set display defaults

plt.rcParams['figure.figsize'] = (10, 10) # large images

plt.rcParams['image.interpolation'] = 'nearest' # don't interpolate: show square pixels

plt.rcParams['image.cmap'] = 'gray' # use grayscale output rather than a (potentially misleading) color heatmap

import sys

import os

caffe_root = './' #指定caffe的根目录

sys.path.insert(0, caffe_root + 'python') #将caffe python接口文件路径添加到python path中

import caffe

# 判断model文件是否存在

if os.path.isfile(caffe_root + 'models/bvlc_reference_caffenet/bvlc_reference_caffenet.caffemodel'):

print 'CaffeNet found.'

else:

print 'Downloading pre-trained CaffeNet model...'

加载网络，创建输入处理

使用python  caffe.io.loadImage接口读取图片，返回的是[0-1]返回的np.float32数组

def load_image(filename, color=True):

"""

Load an image converting from grayscale or alpha as needed.

Parameters

----------

filename : string

color : boolean

flag for color format. True (default) loads as RGB while False

loads as