From: Dmitry Mozolev/AI Tools Lab /SRR/Engineer/삼성전자 <d.mozolev@samsung.com>
Date: Thu, 12 Jul 2018 08:41:54 +0000 (+0300)
Subject: Add Caffe importer example (#592)
X-Git-Tag: nncc_backup~2439
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=edbac4a2f427581960a2bd18fd6da2fdbc43df4f;p=platform%2Fcore%2Fml%2Fnnfw.git

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
--- 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
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;
+}