darknet parser tested

2020-06-01 14:55:08 +02:00
parent d2e2669b6d
commit 18794d52c2
21 changed files with 112 additions and 30 deletions
@@ -8,6 +8,11 @@ int testInference(std::vector<std::string> input_bins, std::vector<std::string>
        if(net->layers[i]->final)
            outputs.push_back(net->layers[i]);
    }
+    // no final layers, set last as output
+    if(outputs.size() == 0) {
+        outputs.push_back(net->layers[net->num_layers-1]);
+    }
+

    // check input
    if(input_bins.size() != 1) {