From 041a5cf65d93b08759f54e4f1b16f1451ca63000 Mon Sep 17 00:00:00 2001 From: Micaela Verucchi Date: Sat, 30 May 2020 18:00:22 +0200 Subject: [PATCH] Add some fields Signed-off-by: Micaela Verucchi --- include/tkDNN/DarknetParser.h | 8 ++++++++ tests/yolo3/yolo3.cpp | 2 ++ 2 files changed, 10 insertions(+) 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<