Add Caffe importer example (#592)

It prints the model and creates Model IR (but doesn't use it).

Signed-off-by: Dmitry Mozolev <d.mozolev@samsung.com>

diff --git a/contrib/nnc/libs/frontend/caffe/CMakeLists.txt b/contrib/nnc/libs/frontend/caffe/CMakeLists.txt
index c78457f..50e6efb 100644 (file)
--- a/contrib/nnc/libs/frontend/caffe/CMakeLists.txt
+++ b/contrib/nnc/libs/frontend/caffe/CMakeLists.txt
@@ -24,3 +24,6 @@ target_link_libraries(caffe_importer PUBLIC ${nn_import_common})
 target_link_libraries(caffe_importer PRIVATE nnc_core)
 target_link_libraries(caffe_importer PRIVATE nncc_core)
 target_link_libraries(caffe_importer PRIVATE nnc_plugin_core)
+
+add_nncc_example_executable(caffe_model_dumper examples/model_dump.cpp)
+nncc_target_link_libraries(caffe_model_dumper caffe_importer)

diff --git a/contrib/nnc/libs/frontend/caffe/examples/model_dump.cpp b/contrib/nnc/libs/frontend/caffe/examples/model_dump.cpp
new file mode 100644 (file)
index 0000000..d0200bb
--- /dev/null
+++ b/contrib/nnc/libs/frontend/caffe/examples/model_dump.cpp
@@ -0,0 +1,32 @@
+#include <iostream>
+
+#include "caffe_importer.h"
+
+int main(int argc, char** argv)
+{
+    std::string modelName;
+    if (argc > 1)
+    {
+        modelName = argv[1];
+    }
+    else
+    {
+        modelName = "mobilenet.caffemodel";
+    }
+
+    nncc::contrib::frontend::caffe::CaffeImporter importer{modelName};
+
+    bool success = importer.import();
+
+    if (success)
+    {
+        importer.dump();
+        importer.createIR();
+    }
+    else
+    {
+        std::cout << "Could not load model \"" << modelName << "\"" << std::endl;
+    }
+
+    return 0;
+}