Add yolov4-csp-swish

2021-09-15 08:24:13 -04:00
parent 000af24d29
commit 2e8ffb9d1d
2 changed files with 1395 additions and 0 deletions
@@ -0,0 +1,40 @@
+//
+// Created by Adam T. Cuellar on 9/15/21.
+//
+
+#include<iostream>
+#include<vector>
+#include "tkdnn.h"
+#include "test.h"
+#include "DarknetParser.h"
+
+int main() {
+    std::string bin_path  = "yolov4-csp-swish";
+    std::vector<std::string> input_bins = {
+            bin_path + "/layers/input.bin"
+    };
+    std::vector<std::string> output_bins = {
+            bin_path + "/debug/layer167_out.bin",
+            bin_path + "/debug/layer171_out.bin",
+            bin_path + "/debug/layer175_out.bin"
+    };
+    std::string wgs_path  = bin_path + "/layers";
+    std::string cfg_path  = std::string(TKDNN_PATH) + "/tests/darknet/cfg/yolov4-csp-swish.cfg";
+    std::string name_path = std::string(TKDNN_PATH) + "/tests/darknet/names/coco.names";
+    downloadWeightsifDoNotExist(input_bins[0], bin_path, "https://cloud.hipert.unimore.it/s/5MFjtNtgbDGdJEo/download");
+
+
+
+    // parse darknet network
+    tk::dnn::Network *net = tk::dnn::darknetParser(cfg_path, wgs_path, name_path);
+    net->print();
+
+    //convert network to tensorRT
+    tk::dnn::NetworkRT *netRT = new tk::dnn::NetworkRT(net, net->getNetworkRTName(bin_path.c_str()));
+
+    int ret = testInference(input_bins, output_bins, net, netRT);
+    net->releaseLayers();
+    delete net;
+    delete netRT;
+    return ret;
+}