common_place Namespace Reference

Functions
	regular_step ()
	save_result (output_path, output_data)
	read_input (input_path)

Function Documentation

read_input()

common_place.read_input

(

input_path

)

Definition at line 64 of file common_place.py.

def read_input(input_path):
    h5f = h5py.File(input_path, 'r')
    for t in h5f:
        tensorName = str(t)
    return h5py.File(input_path, 'r')[tensorName][:]

Referenced by model_runner_caffe.run_caffe(), model_runner_caffe2.run_caffe2(), model_runner_onnx.run_onnx(), and model_runner_tflite.run_tflite().

regular_step()

common_place.regular_step

(

)

This function is intended to decompose the necessary steps to obtain information from the command line.

:return: argparse object, which hold paths to nn model and input data

Definition at line 6 of file common_place.py.

def regular_step():
    """
    This function is intended to decompose the necessary steps to obtain information from the command line.
 
    :return: argparse object, which hold paths to nn model and input data
    """
    parser = argparse.ArgumentParser(formatter_class=RawTextHelpFormatter)
 
    parser.add_argument('-m',
                        '--model',
                        help=("specify input file with NN model, \n[depends from model, "
                              " two for caffe and caffe2, one for onnx and tflite]"),
                        nargs='+')
    parser.add_argument('-i',
                        '--input',
                        help=(" specify file with neural"
                              " network input data, hdf5 for caffe caffe2 tflite "
                              "and pb for onnx"),
                        required=True)
    parser.add_argument(
        '-o',
        '--output_path',
        help='here you specify which place will hold your output, default here',
        default='')
 
    args = parser.parse_args()
    # added to check is our input file or not. most simple way
    try:
        with open(args.input) as f:
            pass
    except IOError as e:
        print('input file your enter doesnt exist!')
 
    # added to check is our model right or not
    try:
        for i in args.model:
            with open(i) as f:
                pass
    except IOError as e:
        print('model you enter doesnt exist, write correct PATH ')
 
    return args
 
 

save_result()

common_place.save_result	(		output_path,
			output_data
	)

This function save result of nn working in .hdf5 file
:param output_path: you specify directory to store your result
:param output_data: information that you write to .hdf5 file
:return:

Definition at line 50 of file common_place.py.

def save_result(output_path, output_data):
    """
    This function save result of nn working in .hdf5 file
    :param output_path: you specify directory to store your result
    :param output_data: information that you write to .hdf5 file
    :return:
    """
    with open(output_path + 'responce.txt', 'w+') as f:
        f.write(str(output_data))
    f = h5py.File(output_path + 'responce.hdf5', 'w')
    f.create_dataset('out', dtype='float32', data=output_data)
    f.close()
 
 

Referenced by model_runner_caffe.run_caffe(), model_runner_caffe2.run_caffe2(), model_runner_onnx.run_onnx(), and model_runner_tflite.run_tflite().