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

YOLO IN TENSORT :)

Browse Source
This commit is contained in:
Francesco Gatti
2017-08-03 16:50:57 +02:00
parent 4e189755cf
commit 0ff47ad6ba
5 changed files with 111 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/yolo/yolo.cpp
+7 -2
View File
@@ -96,7 +96,7 @@ int main() {
tkDNN::Conv2d c29(&net, 1024, 3, 3, 1, 1, 1, 1, c29_bin, true);
tkDNN::Activation a29(&net, tkDNN::ACTIVATION_LEAKY);
tkDNN::Conv2d c30(&net, 425, 1, 1, 1, 1, 0, 0, c30_bin, false);
// tkDNN::Region g31(&net, 80, 4, 5, 0.6f);
tkDNN::Region g31(&net, 80, 4, 5, 0.6f);
// Load input
value_type *data;
@@ -126,7 +126,12 @@ int main() {
}
std::cout<<"\n======= CHECK RESULT =======\n";
std::cout<<"Diffs: "<<checkResult(net.getOutputDim().tot(), out_data, out_data2)<<"\n";
value_type *out, *out_h;
int out_dim = net.getOutputDim().tot();
readBinaryFile(output_bin, out_dim, &out_h, &out);
std::cout<<"CUDNN vs correct Diffs: "<<checkResult(out_dim, out_data, out)<<"\n";
std::cout<<"TRT vs correct Diffs: "<<checkResult(out_dim, out_data2, out)<<"\n";
std::cout<<"CUDNN vs TRT Diffs: "<<checkResult(out_dim, out_data, out)<<"\n";
return 0;
}