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

layer parser

Browse Source
This commit is contained in:
Francesco Gatti
2020-05-30 18:02:20 +02:00
parent 8935d85e91
commit e1d5f58c3f
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/tkDNN/DarknetParser.h
+8
View File
@@ -38,6 +38,14 @@ namespace tk { namespace dnn {
std::cout<<"Add layer: "<<fields.type<<"\n"; std::cout<<"Add layer: "<<fields.type<<"\n";
if(fields.type == "convolutional") { if(fields.type == "convolutional") {
} else if(fields.type == "shortcut") {
} else if(fields.type == "upsample") {
} else if(fields.type == "route") {
} else if(fields.type == "yolo") {
} else{ } else{
FatalError("layer not supported: " + fields.type); FatalError("layer not supported: " + fields.type);
} }