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

yolo4tiny works on tensorRT 🐬 🐬 🐬 🐬

Browse Source 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
This commit is contained in:
Micaela Verucchi
2020-06-30 15:52:58 +02:00
parent 61aa24c6b7
commit fe2e4eae92
4 changed files with 27 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/darknet/yolo4tiny.cpp
+3 -3
View File
@@ -23,11 +23,11 @@ int main() {
net->print();
//convert network to tensorRT
// tk::dnn::NetworkRT *netRT = new tk::dnn::NetworkRT(net, net->getNetworkRTName(bin_path.c_str()));
tk::dnn::NetworkRT *netRT = new tk::dnn::NetworkRT(net, net->getNetworkRTName(bin_path.c_str()));
int ret = testInference(input_bins, output_bins, net, nullptr);
int ret = testInference(input_bins, output_bins, net, netRT);
net->releaseLayers();
delete net;
// delete netRT;
delete netRT;
return ret;
}