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

darknet parser tested

Browse Source
This commit is contained in:
Francesco Gatti
2020-06-01 14:55:08 +02:00
parent 21518c5eef
commit af9bb411be
21 changed files with 112 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/darknet/cfg/yolo2tiny.cfg
+4 -4
View File
@@ -1,5 +1,5 @@
[net]
Training
# Training
batch=64
subdivisions=8
# Testing
@@ -90,9 +90,9 @@ stride=1
pad=1
activation=leaky
#[maxpool]
#size=2
#stride=1
[maxpool]
size=2
stride=1
[convolutional]
batch_normalize=1
tests/darknet/csresnext50-panet-spp.cpp
+1
View File
@@ -27,6 +27,7 @@ int main() {
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;
tests/darknet/csresnext50-panet-spp_berkeley.cpp
+3 -1
View File
@@ -5,7 +5,7 @@
#include "DarknetParser.h"
int main() {
std::string bin_path = "bdd-csresnext50-panet-spp";
std::string bin_path = "csresnext50-panet-spp_berkeley";
std::vector<std::string> input_bins = {
bin_path + "/layers/input.bin"
};
@@ -17,6 +17,7 @@ int main() {
std::string wgs_path = bin_path + "/layers";
std::string cfg_path = "../tests/darknet/cfg/csresnext50-panet-spp_berkeley.cfg";
std::string name_path = "../tests/darknet/names/berkeley.names";
// FIXME: wrong weights
// downloadWeightsifDoNotExist(input_bins[0], bin_path, "https://cloud.hipert.unimore.it/s//download");
// parse darknet network
@@ -27,6 +28,7 @@ int main() {
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;
tests/darknet/yolo2.cpp
+2 -1
View File
@@ -10,7 +10,7 @@ int main() {
bin_path + "/layers/input.bin"
};
std::vector<std::string> output_bins = {
bin_path + "layers/output.bin"
bin_path + "/layers/output.bin"
};
std::string wgs_path = bin_path + "/layers";
std::string cfg_path = "../tests/darknet/cfg/yolo2.cfg";
@@ -25,6 +25,7 @@ int main() {
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;
tests/darknet/yolo2_voc.cpp
+1
View File
@@ -25,6 +25,7 @@ int main() {
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;
tests/darknet/yolo2tiny.cpp
+4 -2
View File
@@ -10,12 +10,13 @@ int main() {
bin_path + "/layers/input.bin"
};
std::vector<std::string> output_bins = {
bin_path + "layers/output.bin"
bin_path + "/layers/output.bin"
};
std::string wgs_path = bin_path + "/layers";
std::string cfg_path = "../tests/darknet/cfg/yolo2tiny.cfg";
std::string name_path = "../tests/darknet/names/coco.names";
downloadWeightsifDoNotExist(input_bins[0], bin_path, "https://cloud.hipert.unimore.it/s/nf4PJ3k8bxBETwL/download");
// FIXME: wrong weights
//downloadWeightsifDoNotExist(input_bins[0], bin_path, "https://cloud.hipert.unimore.it/s//download");
// parse darknet network
tk::dnn::Network *net = tk::dnn::darknetParser(cfg_path, wgs_path, name_path);
@@ -25,6 +26,7 @@ int main() {
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;
tests/darknet/yolo3.cpp
+2 -1
View File
@@ -26,7 +26,8 @@ int main() {
//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);
int ret = testInference(input_bins, output_bins, net, netRT);
net->releaseLayers();
delete net;
delete netRT;
return ret;
tests/darknet/yolo3_512.cpp
+1
View File
@@ -27,6 +27,7 @@ int main() {
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;
tests/darknet/yolo3_berkeley.cpp
+2 -1
View File
@@ -16,7 +16,7 @@ int main() {
};
std::string wgs_path = bin_path + "/layers";
std::string cfg_path = "../tests/darknet/cfg/yolo3_berkeley.cfg";
std::string name_path = "../tests/darknet/names/barkeley.names";
std::string name_path = "../tests/darknet/names/berkeley.names";
downloadWeightsifDoNotExist(input_bins[0], bin_path, "https://cloud.hipert.unimore.it/s/o5cHa4AjTKS64oD/download");
// parse darknet network
@@ -27,6 +27,7 @@ int main() {
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;
tests/darknet/yolo3_coco4.cpp
+1
View File
@@ -27,6 +27,7 @@ int main() {
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;
tests/darknet/yolo3_flir.cpp
+1
View File
@@ -27,6 +27,7 @@ int main() {
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;
tests/darknet/yolo3tiny.cpp
+3 -1
View File
@@ -10,7 +10,8 @@ int main() {
bin_path + "/layers/input.bin"
};
std::vector<std::string> output_bins = {
bin_path + "debug/layer23_out.bin",
bin_path + "/debug/layer16_out.bin",
bin_path + "/debug/layer23_out.bin",
};
std::string wgs_path = bin_path + "/layers";
std::string cfg_path = "../tests/darknet/cfg/yolo3tiny.cfg";
@@ -25,6 +26,7 @@ int main() {
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;
tests/darknet/yolo3tiny512.cpp → tests/darknet/yolo3tiny_512.cpp
+3 -1
View File
@@ -10,7 +10,8 @@ int main() {
bin_path + "/layers/input.bin"
};
std::vector<std::string> output_bins = {
bin_path + "debug/layer23_out.bin",
bin_path + "/debug/layer16_out.bin",
bin_path + "/debug/layer23_out.bin",
};
std::string wgs_path = bin_path + "/layers";
std::string cfg_path = "../tests/darknet/cfg/yolo3tiny_512.cfg";
@@ -25,6 +26,7 @@ int main() {
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;
tests/darknet/yolo4.cpp
+1
View File
@@ -27,6 +27,7 @@ int main() {
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;
tests/darknet/yolo4_berkeley.cpp
+34
View File
@@ -0,0 +1,34 @@
#include<iostream>
#include<vector>
#include "tkdnn.h"
#include "test.h"
#include "DarknetParser.h"
int main() {
std::string bin_path = "yolo4_berkeley";
std::vector<std::string> input_bins = {
bin_path + "/layers/input.bin"
};
std::vector<std::string> output_bins = {
bin_path + "/debug/layer139_out.bin",
bin_path + "/debug/layer150_out.bin",
bin_path + "/debug/layer161_out.bin"
};
std::string wgs_path = bin_path + "/layers";
std::string cfg_path = "../tests/darknet/cfg/yolo4_berkeley.cfg";
std::string name_path = "../tests/darknet/names/berkeley.names";
downloadWeightsifDoNotExist(input_bins[0], bin_path, "https://cloud.hipert.unimore.it/s//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;
}