Page History

def mxnet.contrib.onnx.get_model_metadata(model_file):
	"""
	Returns the metadata information of the given model. Currently it will return the shape information of the input and output tensors.
	Input Parameters –
	----------
	model_file – 
		A string object representing the path to the ONNX model file.
 
	Return Type –
	----------
	model_metadata – 
		A dictionary object mapping various metadata to its corresponding value. The proposed implementation will return a dictionary with following template.
		{
			“input_tensor_shape”: <list of tuples representing the shape of the input paramters>,
			“output_tensor_shape”: <list of tuples representing the shape of the output of the model>
		}
	"""
	...
	return metadata