model_runner_onnx Namespace Reference

Functions
	run_onnx (model, input_path, output_path='')

Variables
	args = regular_step()

Function Documentation

run_onnx()

model_runner_onnx.run_onnx	(		model,
			input_path,
			output_path = ''
	)

Definition at line 7 of file model_runner_onnx.py.

def run_onnx(model, input_path, output_path=''):  #args.model[0] , args.input
    path = model
 
    #I'll leave it in case anyone needs to read the .pb file.
    #proto_arr = onnx.TensorProto()
    #with open(input_path, 'rb') as f:
    #    proto_arr.ParseFromString(f.read())
    #    input_arr = onnx.numpy_helper.to_array(proto_arr)
 
    modelFile = onnx.load(path, 'rb')
    input_arr = read_input(input_path)
    output = caffe2.python.onnx.backend.run_model(modelFile, input_arr)
 
    print(output)
    save_result(output_path, output)
 
 

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

Variable Documentation

args

model_runner_onnx.args = regular_step()

Definition at line 25 of file model_runner_onnx.py.