mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity

Add some fields

Browse Source 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
This commit is contained in:
Micaela Verucchi
2020-05-30 18:00:22 +02:00
parent 98173ec15c
commit 26e6ba52a5
2 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/tkDNN/DarknetParser.h
+8
View File
@@ -59,6 +59,14 @@ namespace tk { namespace dnn {
fields.batch_normalize = std::stoi(value); fields.batch_normalize = std::stoi(value);
else if (name == "filters") else if (name == "filters")
fields.filters = std::stoi(value); 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; return true;
} }
tests/yolo3/yolo3.cpp
+2
View File
@@ -10,6 +10,8 @@ int main() {
tk::dnn::darknetFields_t f; tk::dnn::darknetFields_t f;
tk::dnn::darknetParseFields("width=40", f); tk::dnn::darknetParseFields("width=40", f);
tk::dnn::darknetParseFields("height=40", f); tk::dnn::darknetParseFields("height=40", f);
tk::dnn::darknetParseFields("channels=40", f);
tk::dnn::darknetParseFields("activation=leaky", f);
std::cout<<f<<std::endl; std::cout<<f<<std::endl;
// Network layout // Network layout