欢迎使用CSDN-markdown编辑器

import os
import sys
import numpy as np
import matplotlib.pyplot as plt

caffe_root = ‘home/git/公共的/caffe’
sys.path.insert(0,caffe_root + ‘python’)

import caffe
MODEL_FILE = ‘home/git//caffe/examples/mnist/lenet.prototxt’
PRETRAINED = ‘home/git/usr/caffe/examples/mnist/lenet_iter_10000.caffemodel’
IMAGE_FILE = ‘home/git/usr/caffe/examples/images/cat.jpg’

input_image = caffe.io.load_image(IMAGE_FILE,color=False)

print input_image

net = caffe.Classifier(MODEL_FILE,PRETRAINED,channel_swap=(2,1,0),image_dims(28,28))

net = caffe.Classifier(MODEL_FILE,PRETRAINED)

prdediction = net.predict([input_image],oversample=False)
caffe.set_model_cpu()
print ‘predicted class:’.prediction[0].argmax()

公共的

CPU_ONLY := 1BLAS := atlasPYTHON_INCLUDE := /usr/include/python2.7 \        /usr/lib/python2.7/dist-packages/numpy/core/includeANACONDA_HOME := $(HOME)/anacondaPYTHON_INCLUDE := $(ANACONDA_HOME)/include \         $(ANACONDA_HOME)/include/python2.7 \         $(ANACONDA_HOME)/lib/python2.7/site-packages/numpy/core/includePYTHON_LIB := $(ANACONDA_HOME)/libWITH_PYTHON_LAYER := 1INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/includeLIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/libBUILD_DIR := buildDISTRIBUTE_DIR := distributeTEST_GPUID := 0Q ?= @