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

get regions

Browse Source
This commit is contained in:
Francesco Gatti
2017-08-08 17:17:24 +02:00
parent 0a9957ba18
commit d7ce952465
4 changed files with 206 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/yolo-tiny/yolo-tiny.cpp
+5 -1
View File
@@ -12,6 +12,7 @@ const char *c10_bin = "../tests/yolo-tiny/layers/c10.bin";
const char *c12_bin = "../tests/yolo-tiny/layers/c12.bin";
const char *c13_bin = "../tests/yolo-tiny/layers/c13.bin";
const char *c14_bin = "../tests/yolo-tiny/layers/c14.bin";
const char *g15_bin = "../tests/yolo-tiny/layers/g15.bin";
const char *output_bin = "../tests/yolo-tiny/layers/outputLEL.bin";
int main() {
@@ -49,7 +50,7 @@ int main() {
tkDNN::Conv2d c13(&net, 1024, 3, 3, 1, 1, 1, 1, c13_bin, true);
tkDNN::Activation a13(&net, tkDNN::ACTIVATION_LEAKY);
tkDNN::Conv2d c14(&net, 125, 1, 1, 1, 1, 0, 0, c14_bin, false);
tkDNN::Region g15(&net, 20, 4, 5, 0.6f);
tkDNN::Region g15(&net, 20, 4, 5, 0.6f, g15_bin);
// Load input
value_type *data;
@@ -86,5 +87,8 @@ 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";
g15.interpretData();
return 0;
}
tests/yolo/yolo.cpp
+5 -1
View File
@@ -25,6 +25,7 @@ const char *c24_bin = "../tests/yolo/layers/c24.bin";
const char *c26_bin = "../tests/yolo/layers/c26.bin";
const char *c29_bin = "../tests/yolo/layers/c29.bin";
const char *c30_bin = "../tests/yolo/layers/c30.bin";
const char *g31_bin = "../tests/yolo/layers/g31.bin";
const char *output_bin = "../tests/yolo/layers/output.bin";
int main() {
@@ -96,7 +97,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, g31_bin);
// Load input
value_type *data;
@@ -133,5 +134,8 @@ 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";
g31.interpretData();
return 0;
}