From cb6dc4f1826ad941fdfa07396b13f13db562a713 Mon Sep 17 00:00:00 2001
From: Micaela Verucchi <micaelaverucchi@gmail.com>
Date: Sat, 30 May 2020 17:25:48 +0200
Subject: [PATCH] Add parseType

Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
---
 include/tkDNN/DarknetParser.h | 13 ++++++++++---
 tests/yolo3/yolo3.cpp         |  1 +
 2 files changed, 11 insertions(+), 3 deletions(-)

diff --git a/include/tkDNN/DarknetParser.h b/include/tkDNN/DarknetParser.h
index bc1a8da..c07d813 100644
--- a/include/tkDNN/DarknetParser.h
+++ b/include/tkDNN/DarknetParser.h
@@ -9,9 +9,16 @@ namespace tk { namespace dnn {
         tk::dnn::dataDim_t dim;
         tk::dnn::Network *net = new tk::dnn::Network(dim);
 
-
+    }
+        
+    std::string parseType(const std::string& line){
+            size_t start = line.find("[");
+            size_t end  = line.find("]");
+            if( start == std::string::npos || end == std::string::npos)
+                return "";
+            start++;
+            std::string type = line.substr(start, end-start);
+            return type;
     }
 
-
-
 }}
diff --git a/tests/yolo3/yolo3.cpp b/tests/yolo3/yolo3.cpp
index 324a4c6..9187a54 100644
--- a/tests/yolo3/yolo3.cpp
+++ b/tests/yolo3/yolo3.cpp
@@ -6,6 +6,7 @@
 int main() {
 
     tk::dnn::Network *net = tk::dnn::DarknetParser("../../tests/yolo3/yolo3.cfg");
+    tk::dnn::parseType("[net]");
 
     // Network layout
     //tk::dnn::dataDim_t dim(1, 3, 416, 416, 1);