model_runner_caffe Namespace Reference

Functions
	run_caffe (model_topology, model_weight, input_path, output_path='')

Variables
	args = regular_step()

Function Documentation

run_caffe()

model_runner_caffe.run_caffe	(		model_topology,
			model_weight,
			input_path,
			output_path = ''
	)

Definition at line 5 of file model_runner_caffe.py.

def run_caffe(model_topology, model_weight, input_path, output_path=''):
    path = model_topology
    path_w = model_weight
 
    net = caffe.Net(path_w, path, caffe.TEST)
    # TODO get 'data' parameter more universal, blobs contain other names
    net.blobs['data'].data[...] = read_input(input_path)
    out = net.forward()
    all_names = [n for n in net._layer_names]
    out = out[all_names[-1]]
    save_result(output_path, out)
    print(out)
 
 

References common_place.read_input(), and common_place.save_result().

Variable Documentation

args

model_runner_caffe.args = regular_step()

Definition at line 20 of file model_runner_caffe.py.