diff --git a/include/tkDNN/DarknetParser.h b/include/tkDNN/DarknetParser.h index c921d2c..76be28c 100644 --- a/include/tkDNN/DarknetParser.h +++ b/include/tkDNN/DarknetParser.h @@ -5,9 +5,10 @@ namespace tk { namespace dnn { struct darknetFields_t{ + std::string type = ""; int width = 0; int height = 0; - + int channels = 3; }; std::string darknetParseType(const std::string& line){ @@ -20,21 +21,37 @@ namespace tk { namespace dnn { return type; } - darknetFields_t parseFields(const std::string& line){ - + bool darknetParseFields(const std::string& line, darknetFields_t &fields){ + return true; } + tk::dnn::Network *darknetAddNet(darknetFields_t &fields) { + std::cout<<"Add Net: "< 0) { - std::cout<<"type: "<