here I write how I run model on my computer
sections:
a) goal of this script
b) examples of code running in author’s local machine
c) parametrs and short comment
goal of this script
Here the author has attempted to implement a program capable of running any of the 4 models (caffe, caffe2, tflite, onnx) in a simple and user-friendly manner. The goal of the program is to get the file containing the output of the computation graph at the program output.
examples of code running in author’s local machine
The purpose of the examples below is to demonstrate which arguments and in which order you should use to run this script correctly.
caffe:
$ python3 model_runner.py -m caffe1_runer/inception-v3_ref.caffemodel caffe1_runer/inception-v3_ref.prototxt -i caffe1_runer/ILSVRC2012_val_00000002.JPEG.tfl.hdf5
caffe2:
$ python model_runner.py -m caffe2_runer_and_photo/caffe2_models/init_net.pb caffe2_runer_and_photo/caffe2_models/predict_net.pb -i randomInput.hdf5
tflite:
$ python model_runner.py -m tflite_runer_and_photo/TST-1-2\ AVARAGE_POOP_2D.tflite -i tflite_runer_and_photo/in.hdf5
onnx:
$ python model_runner.py -m onnx_runer/model.onnx -i RANDOM.hdf5