安装好caffe之后在官方网站下载model和script和image。分别代表模型,直接使用的脚本和测试图像。
放到/caffe/DRIU/
中。
打开test.py
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.cm as cm
import scipy.misc
from PIL import Image
import scipy.io
import os
import scipy
import sys
# Point to Caffe folder /path/to/caffe
caffe_root = '../'
# Choose between 'DRIVE', 'STARE', 'DRIONS', and 'RIMONE'
database = 'DRIVE'
# Use GPU?这里将use_gpu设置为1,将gpu_id设置为0,caffe.set_device的gpu_id从0开始计数。
use_gpu = 1;
gpu_id = 'gpu 0';
#sys.path.insert(0, caffe_root+'python/') 这里注释掉了,因为ms的caffe的根目录不在python下,在build\x64\release\pycaffe\caffe下。由于我将caffe的根目录直接放在python的sitepackage下,所以不用这句话直接使用import caffe也没问题。
import caffe
os.chdir(caffe_root+'/DRIU/')
def imshow_im(im):
plt.imshow(im,interpolation='none',cmap=cm.Grays_r)
net_struct = 'deploy_'+database+'.prototxt'
data_root = caffe_root+'/DRIU/Images/'+database+'/'
save_root = caffe_root+'/DRIU/results/'+database+'/'
if not os.path.exists(save_root):
os.makedirs(save_root)
with open(data_ro