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

fix wrong commit

Browse Source
This commit is contained in:
Francesco Gatti
2020-06-14 13:03:48 +02:00
parent cbfc8ea4f2
commit 6dff675db7
1 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/darknet/yolo3.cpp
+8 -8
View File
@@ -5,19 +5,19 @@
#include "DarknetParser.h" #include "DarknetParser.h"
int main() { int main() {
std::string bin_path = "yolov3-sppx"; std::string bin_path = "yolo3";
std::vector<std::string> input_bins = { std::vector<std::string> input_bins = {
bin_path + "/layers/input.bin" bin_path + "/layers/input.bin"
}; };
std::vector<std::string> output_bins = { std::vector<std::string> output_bins = {
bin_path + "/debug/layer89_out.bin", bin_path + "/debug/layer82_out.bin",
bin_path + "/debug/layer101_out.bin", bin_path + "/debug/layer94_out.bin",
bin_path + "/debug/layer113_out.bin" bin_path + "/debug/layer106_out.bin"
}; };
std::string wgs_path = bin_path + "/layers"; std::string wgs_path = bin_path + "/layers";
std::string cfg_path = "../tests/darknet/cfg/yolov3-sppx.cfg"; std::string cfg_path = "../tests/darknet/cfg/yolo3.cfg";
std::string name_path = "../tests/darknet/names/coco4.names"; std::string name_path = "../tests/darknet/names/coco.names";
//downloadWeightsifDoNotExist(input_bins[0], bin_path, "https://cloud.hipert.unimore.it/s/jPXmHyptpLoNdNR/download"); downloadWeightsifDoNotExist(input_bins[0], bin_path, "https://cloud.hipert.unimore.it/s/jPXmHyptpLoNdNR/download");
// parse darknet network // parse darknet network
tk::dnn::Network *net = tk::dnn::darknetParser(cfg_path, wgs_path, name_path); tk::dnn::Network *net = tk::dnn::darknetParser(cfg_path, wgs_path, name_path);
@@ -31,4 +31,4 @@ int main() {
delete net; delete net;
delete netRT; delete netRT;
return ret; return ret;
} }