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

Add fields to darknetParseFields

Browse Source 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
This commit is contained in:
Micaela Verucchi
2020-05-30 18:44:17 +02:00
parent 6b6dbabba3
commit e5e6654b1d
2 changed files with 56 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/yolo3/yolo3.cpp
+3 -3
View File
@@ -8,10 +8,10 @@ int main() {
tk::dnn::Network *net = tk::dnn::DarknetParser("../tests/yolo3/yolov3.cfg");
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("channels=40", f);
tk::dnn::darknetParseFields("activation=leaky", f);
tk::dnn::darknetParseFields("layers = -1", f);
tk::dnn::darknetParseFields("mask = 3,4,5", f);
std::cout<<f<<std::endl;
// Network layout