caffe python接口使用记录

import caffe
import cv2
import numpy as np

def load_data(image_path):
    image = cv2.imread(image_path)
    image = cv2.resize(image, (448,448))
    image = (image-127.5)/127.5
    image = image[:,:,[2,1,0]]
    return image

#caffe.set_mode_cpu()
caffe.set_mode_gpu()
net_path = "deploy.prototxt"
model_path = "models.caffemodel"

net = caffe.Net(net_path, model_path, caffe.TEST)

input = net.blobs[net.inputs[0]]
image_path = "dog.jpg"
input_data = load_data(image_path)
#input.data[...] = input
input_data_r = input_data[:,:,0]
input_data_g = input_data[:,:,1]
input_data_b = input_data[:,:,2]

for i in range(448):
    for j in range(448):
        input.data[0,0,i,j] = input_data_r[i,j]
        input.data[0,1,i,j] = input_data_g[i,j]
        input.data[0,2,i,j] = input_data_b[i,j]
net._forward(0,len(net.layers)-1)
output = net.blobs[net.outputs[0]]