diff --git a/include/tkDNN/DarknetParser.h b/include/tkDNN/DarknetParser.h index 09d8dce..d8e8b59 100644 --- a/include/tkDNN/DarknetParser.h +++ b/include/tkDNN/DarknetParser.h @@ -59,6 +59,14 @@ namespace tk { namespace dnn { fields.batch_normalize = std::stoi(value); else if (name == "filters") fields.filters = std::stoi(value); + else if (name == "size") + fields.size = std::stoi(value); + else if (name == "stride") + fields.stride = std::stoi(value); + else if (name == "pad") + fields.pad = std::stoi(value); + else if (name == "activation") + fields.activation = value; return true; } diff --git a/tests/yolo3/yolo3.cpp b/tests/yolo3/yolo3.cpp index f74faa5..f1602da 100644 --- a/tests/yolo3/yolo3.cpp +++ b/tests/yolo3/yolo3.cpp @@ -10,6 +10,8 @@ int main() { tk::dnn::darknetFields_t f; tk::dnn::darknetParseFields("width=40", f); tk::dnn::darknetParseFields("height=40", f); + tk::dnn::darknetParseFields("channels=40", f); + tk::dnn::darknetParseFields("activation=leaky", f); std::cout<