third_party/opencv
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added tests for ResNet-50 and SqueezeNet v1.1

Browse Source
This commit is contained in:
Aleksandr Rybnikov
2017-07-03 16:29:30 +03:00
parent d5323ce848
commit d3c14ce3ea
2 changed files with 35 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/dnn/test/test_halide_nets.cpp
+4 -4
View File
@@ -104,8 +104,8 @@ TEST(Reproducibility_ResNet_50_Halide, Accuracy)
TEST(Reproducibility_SqueezeNet_v1_1_Halide, Accuracy)
{
test(findDataFile("dnn/squeezenet_v1_1.caffemodel", false),
findDataFile("dnn/squeezenet_v1_1.prototxt", false),
test(findDataFile("dnn/squeezenet_v1.1.caffemodel", false),
findDataFile("dnn/squeezenet_v1.1.prototxt", false),
findDataFile("dnn/halide_scheduler_squeezenet_v1_1.yml", false),
227, 227, "prob", "caffe", DNN_TARGET_CPU);
};
@@ -151,8 +151,8 @@ TEST(Reproducibility_ResNet_50_Halide_opencl, Accuracy)
TEST(Reproducibility_SqueezeNet_v1_1_Halide_opencl, Accuracy)
{
test(findDataFile("dnn/squeezenet_v1_1.caffemodel", false),
findDataFile("dnn/squeezenet_v1_1.prototxt", false),
test(findDataFile("dnn/squeezenet_v1.1.caffemodel", false),
findDataFile("dnn/squeezenet_v1.1.prototxt", false),
findDataFile("dnn/halide_scheduler_opencl_squeezenet_v1_1.yml", false),
227, 227, "prob", "caffe", DNN_TARGET_OPENCL);
};