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

interpret

Browse Source
This commit is contained in:
Francesco Gatti
2017-08-22 01:32:59 -07:00
parent 5a52de17eb
commit 5595b8037b
8 changed files with 106 additions and 60 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/yolo-tiny/yolo-tiny.cpp
+1 -5
View File
@@ -49,7 +49,7 @@ int main() {
tkDNN::Conv2d c12(&net, 512, 3, 3, 1, 1, 1, 1, c12_bin, true);
tkDNN::Activation a12(&net, tkDNN::ACTIVATION_LEAKY);
tkDNN::Conv2d c13(&net, 425, 1, 1, 1, 1, 0, 0, c13_bin, false);
tkDNN::Region g14(&net, 80, 4, 5, 0.6f, g14_bin);
tkDNN::Region g14(&net, 80, 4, 5);
// Load input
dnnType *data;
@@ -89,9 +89,5 @@ int main() {
std::cout<<"CUDNN vs correct"; checkResult(out_dim, out_data, out);
std::cout<<"TRT vs correct"; checkResult(out_dim, out_data2, out);
std::cout<<"CUDNN vs TRT "; checkResult(out_dim, out_data, out_data2);
std::cout<<"\n\nDetected objects: \n";
g14.interpretData();
g14.showImageResult(input_h);
return 0;
}