caffe-python

来源：互联网发布：阿里云ecs ddos 编辑：程序博客网时间：2024/05/22 21:20

用python调用修改caffe网络

import numpy as npimport matplotlib.pyplot as plt%matplotlib inline# 加载caffe路径:caffe_root = '../'  # {caffe_root}/examplesimport syssys.path.insert(0, caffe_root + 'python')import caffe# configure plottingplt.rcParams['figure.figsize'] = (10, 10)plt.rcParams['image.interpolation'] = 'nearest'plt.rcParams['image.cmap'] = 'gray'

设置自己的滤波器：

caffe.set_mode_cpu()#caffe.set_mode_gpu()设置是否使用GPUnet = caffe.Net('net_surgery/conv.prototxt', caffe.TEST)#加载网络结构，并设置是测试还是训练print("blobs {}\nparams {}".format(net.blobs.keys(), net.params.keys()))# load image and prepare as a single input batch for Caffeim = np.array(caffe.io.load_image('images/cat_gray.jpg', color=False)).squeeze()#加载图像转换成np矩阵格式plt.title("original image")plt.imshow(im)plt.axis('off')im_input = im[np.newaxis, np.newaxis, :, :]net.blobs['data'].reshape(*im_input.shape)net.blobs['data'].data[...] = im_input

阅读全文

0 0