[application] Enable draw classification app

Enable draw classification application.
Corresponding bug fixes are also added.

Signed-off-by: Parichay Kapoor <kparichay@gmail.com>

diff --git a/Applications/meson.build b/Applications/meson.build
index 35729fe..5c5a9e7 100644 (file)
--- a/Applications/meson.build
+++ b/Applications/meson.build
@@ -11,7 +11,7 @@ subdir('VGG/jni')
 subdir('Resnet/jni')
 subdir('ReinforcementLearning/DeepQ/jni')
 subdir('TransferLearning/CIFAR_Classification/jni')
-# subdir('TransferLearning/Draw_Classification/jni')
+subdir('TransferLearning/Draw_Classification/jni')
 # subdir('Custom')
 subdir('ProductRatings/jni')
 

diff --git a/nntrainer/graph/network_graph.cpp b/nntrainer/graph/network_graph.cpp
index 8f1f394..acc6fd9 100644 (file)
--- a/nntrainer/graph/network_graph.cpp
+++ b/nntrainer/graph/network_graph.cpp
@@ -802,7 +802,7 @@ int NetworkGraph::initialize(std::shared_ptr<Manager> manager) {
 
       unsigned int j = 0;
       for (; j < out_layer_node->getNumInputConnections(); ++j) {
-        if (out_layer_node->getInputLayers()[j] == lnode->getName()) {
+        if (istrequal(out_layer_node->getInputLayers()[j], lnode->getName())) {
           break;
         }
       }

diff --git a/test/unittest/layers/unittest_layers_tflite.cpp b/test/unittest/layers/unittest_layers_tflite.cpp
index 34f7d71..3091df1 100644 (file)
--- a/test/unittest/layers/unittest_layers_tflite.cpp
+++ b/test/unittest/layers/unittest_layers_tflite.cpp
@@ -18,7 +18,7 @@
 
 auto semantic_tflite = LayerSemanticsParamType(
   nntrainer::createLayer<nntrainer::TfLiteLayer>, nntrainer::TfLiteLayer::type,
-  {"modelfile=../test/test_models/models/add.tflite"}, {}, 0, false);
+  {"modelfile=../test/test_models/models/add.tflite"}, 0, false);
 
 INSTANTIATE_TEST_CASE_P(TfLite, LayerSemantics,
                         ::testing::Values(semantic_tflite));