From d3c14ce3ea31398d6ea5abe7126124936d334685 Mon Sep 17 00:00:00 2001 From: Aleksandr Rybnikov Date: Mon, 3 Jul 2017 16:29:30 +0300 Subject: [PATCH] Added tests for ResNet-50 and SqueezeNet v1.1 --- modules/dnn/test/test_caffe_importer.cpp | 31 ++++++++++++++++++++++++ modules/dnn/test/test_halide_nets.cpp | 8 +++--- 2 files changed, 35 insertions(+), 4 deletions(-) diff --git a/modules/dnn/test/test_caffe_importer.cpp b/modules/dnn/test/test_caffe_importer.cpp index 75c6bb137b..7fe7f1dc74 100644 --- a/modules/dnn/test/test_caffe_importer.cpp +++ b/modules/dnn/test/test_caffe_importer.cpp @@ -157,4 +157,35 @@ TEST(Reproducibility_SSD, Accuracy) Mat ref = blobFromNPY(_tf("ssd_out.npy")); normAssert(ref, out); } + +TEST(Reproducibility_ResNet50, Accuracy) +{ + Net net = readNetFromCaffe(findDataFile("dnn/ResNet-50-deploy.prototxt", false), + findDataFile("dnn/ResNet-50-model.caffemodel", false)); + + Mat input = blobFromImage(imread(_tf("googlenet_0.png")), 1, Size(224,224)); + ASSERT_TRUE(!input.empty()); + + net.setInput(input); + Mat out = net.forward(); + + Mat ref = blobFromNPY(_tf("resnet50_prob.npy")); + normAssert(ref, out); +} + +TEST(Reproducibility_SqueezeNet_v1_1, Accuracy) +{ + Net net = readNetFromCaffe(findDataFile("dnn/squeezenet_v1.1.prototxt", false), + findDataFile("dnn/squeezenet_v1.1.caffemodel", false)); + + Mat input = blobFromImage(imread(_tf("googlenet_0.png")), 1, Size(227,227)); + ASSERT_TRUE(!input.empty()); + + net.setInput(input); + Mat out = net.forward(); + + Mat ref = blobFromNPY(_tf("squeezenet_v1.1_prob.npy")); + normAssert(ref, out); +} + } diff --git a/modules/dnn/test/test_halide_nets.cpp b/modules/dnn/test/test_halide_nets.cpp index fa5553966b..c1ac2ff0c0 100644 --- a/modules/dnn/test/test_halide_nets.cpp +++ b/modules/dnn/test/test_halide_nets.cpp @@ -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); };