caffe实战,deep retinal image understanding 实现

安装好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
  • 1
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值