model_runner_caffe2 Namespace Reference

Functions
	run_caffe2 (init_net, predict_net, input_path, output_path='')

Variables
	args = regular_step()

Function Documentation

run_caffe2()

model_runner_caffe2.run_caffe2	(		init_net,
			predict_net,
			input_path,
			output_path = ''
	)

Definition at line 6 of file model_runner_caffe2.py.

def run_caffe2(init_net, predict_net, input_path, output_path=''):
    x = read_input(input_path)
    with open(init_net, 'rb') as f:
        init_net = f.read()
 
    with open(predict_net, 'rb') as f:
        predict_net = f.read()
    p = workspace.Predictor(init_net, predict_net)
    # TODO get 'data' parameter more universal, blobs contain other names
    results = p.run({'data': x})
    print(results)
    save_result(output_path, results)
 
 

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

Variable Documentation

args

model_runner_caffe2.args = regular_step()

Definition at line 21 of file model_runner_caffe2.py.