[opencv] 13/251: Added tests for ResNet-50 and SqueezeNet v1.1
Nobuhiro Iwamatsu
iwamatsu at moszumanska.debian.org
Sun Aug 27 23:27:18 UTC 2017
This is an automated email from the git hooks/post-receive script.
iwamatsu pushed a commit to annotated tag 3.3.0
in repository opencv.
commit d3c14ce3ea31398d6ea5abe7126124936d334685
Author: Aleksandr Rybnikov <arrybn at gmail.com>
Date: Mon Jul 3 16:29:30 2017 +0300
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 75c6bb1..7fe7f1d 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 fa55539..c1ac2ff 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);
};
--
Alioth's /usr/local/bin/git-commit-notice on /srv/git.debian.org/git/debian-science/packages/opencv.git
More information about the debian-science-commits
mailing list