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

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);