Python boundary_eval示例

示例#1

文件： solve.py 项目： tinyloop/caffe-ccnn

import sys
sys.path.append('../../python')

import caffe
from caffe import surgery, score

import numpy as np
import os

weights = '/home/shelhamer/dpb/exp/vgg16fc.caffemodel'

caffe.set_phase_train()
caffe.set_mode_gpu()
caffe.set_device(1)

solver = caffe.SGDSolver('solver.prototxt')
solver.net.copy_from(weights)

interp_layers = [k for k in solver.net.params.keys() if 'up' in k]
surgery.interp(solver.net, interp_layers)

for _ in range(30):
    print 'epoch {}'.format(_)
    solver.step(1087)
    score.boundary_eval(solver, ('bsds', 'val'), layer='prob')

示例#2

import sys
sys.path.append('../../python')

import caffe
from caffe import surgery, score

import numpy as np
import os

weights = sys.argv[1]

caffe.set_phase_train()
caffe.set_mode_gpu()
caffe.set_device(1)

solver = caffe.SGDSolver('solver.prototxt')
solver.net.copy_from(weights)

score.boundary_eval(solver, 'val', layer='prob')