1 files changed, 22 insertions, 0 deletions
diff --git a/compiler/nnc/utils/model_runner/model_runner_caffe.py b/compiler/nnc/utils/model_runner/model_runner_caffe.py
new file mode 100755
index 000000000..a2e94272c
--- /dev/null
+++ b/compiler/nnc/utils/model_runner/model_runner_caffe.py
@@ -0,0 +1,22 @@
+from common_place import *
+import caffe
+
+
+def run_caffe(model_topology, model_weight, input_path, output_path=''):
+    path = model_topology
+    path_w = model_weight
+
+    net = caffe.Net(path_w, path, caffe.TEST)
+    # TODO get 'data' parameter more universal, blobs contain other names
+    net.blobs['data'].data[...] = read_input(input_path)
+    out = net.forward()
+    all_names = [n for n in net._layer_names]
+    out = out[all_names[-1]]
+    save_result(output_path, out)
+    print(out)
+
+
+if __name__ == '__main__':
+    args = regular_step()
+
+    run_caffe(args.model[0], args.model[1], args.input, args.output_path)