Skip to content
Snippets Groups Projects
OpenCVBenchmark.cpp 1.67 KiB
Newer Older
zhanghb97's avatar
zhanghb97 committed
//===- OpenCVBenchmark.cpp ------------------------------------------------===//
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//
//===----------------------------------------------------------------------===//
//
// This file implements the benchmark for OpenCV.
//
//===----------------------------------------------------------------------===//

#include "ImageProcessing/Kernels.h"
#include <benchmark/benchmark.h>
#include <opencv2/opencv.hpp>

using namespace cv;
using namespace std;

// Read input image and specify kernel.
Mat inputImage = imread("../../benchmarks/ImageProcessing/Images/YuTu.png",
                        IMREAD_GRAYSCALE);
Mat kernelOpencv = Mat(3, 3, CV_32FC1, laplacianKernelAlign);

// Declare output image.
Mat outputOpencv;

// Benchmarking function.
static void BM_OpenCV(benchmark::State &state) {
  for (auto _ : state) {
    for (int i = 0; i < state.range(0); ++i) {
      filter2D(inputImage, outputOpencv, CV_32FC1, kernelOpencv);
    }
  }
}

// Register benchmarking function with different arguments.
BENCHMARK(BM_OpenCV)->Arg(1);
BENCHMARK(BM_OpenCV)->Arg(2);
BENCHMARK(BM_OpenCV)->Arg(4);
BENCHMARK(BM_OpenCV)->Arg(8);
BENCHMARK(BM_OpenCV)->Arg(16);