chore: add more opencv benches

Author: not-matthias

CMakeLists.txt

@@ -2,7 +2,16 @@ cmake_minimum_required(VERSION 3.12)
 project( opencv-benchmarks VERSION 0.0.0 LANGUAGES CXX)
 find_package( OpenCV REQUIRED )
 include_directories( ${OpenCV_INCLUDE_DIRS} )
-add_executable( cv-bench src/main.cpp src/classifier.cpp)
+add_executable(
+    cv-bench
+    src/main.cpp
+    src/classifier.cpp
+    src/color_conversion.cpp
+    src/edge_detection.cpp
+    src/filter.cpp
+    src/read_image.cpp
+    src/transform.cpp
+)
 target_link_libraries( cv-bench ${OpenCV_LIBS} )
 
 
@@ -18,4 +27,3 @@ FetchContent_Declare(
 )
 FetchContent_MakeAvailable(google_benchmark)
 target_link_libraries(cv-bench benchmark::benchmark)
-

src/color_conversion.cpp

@@ -0,0 +1,15 @@
+#include <benchmark/benchmark.h>
+#include <opencv2/opencv.hpp>
+
+static void BM_ColorConversion(benchmark::State& state) {
+  static cv::Mat img = cv::imread("assets/dog_bike_car.jpg");
+  if (img.empty())
+    throw std::runtime_error("Cannot open image");
+
+  for (auto _ : state) {
+    cv::Mat gray;
+    cv::cvtColor(img, gray, cv::COLOR_BGR2GRAY);
+    benchmark::DoNotOptimize(gray.data);
+  }
+}
+BENCHMARK(BM_ColorConversion);

src/edge_detection.cpp

@@ -0,0 +1,19 @@
+#include <benchmark/benchmark.h>
+#include <opencv2/opencv.hpp>
+
+static void BM_Canny(benchmark::State& state) {
+  static cv::Mat img = cv::imread("assets/dog_bike_car.jpg");
+  if (img.empty())
+    throw std::runtime_error("Cannot open image");
+
+  double low = state.range(0);
+  double high = state.range(1);
+
+  for (auto _ : state) {
+    cv::Mat edges;
+    cv::Canny(img, edges, low, high);
+    benchmark::DoNotOptimize(edges.data);
+  }
+}
+
+BENCHMARK(BM_Canny)->Args({50, 150})->Args({75, 175})->Args({100, 200});

src/filter.cpp

@@ -0,0 +1,62 @@
+#include <benchmark/benchmark.h>
+#include <opencv2/opencv.hpp>
+
+// Shared image for all filters in this file
+static cv::Mat GetImage() {
+  static cv::Mat img = cv::imread("assets/dog_bike_car.jpg");
+  if (img.empty())
+    throw std::runtime_error("Cannot open image");
+  return img;
+}
+
+static void BM_GaussianBlur(benchmark::State& state) {
+  cv::Mat img = GetImage();
+  int kernel_size = state.range(0);
+
+  for (auto _ : state) {
+    cv::Mat blurred;
+    cv::GaussianBlur(img, blurred, {kernel_size, kernel_size}, 0);
+    benchmark::DoNotOptimize(blurred.data);
+  }
+}
+
+static void BM_MedianBlur(benchmark::State& state) {
+  cv::Mat img = GetImage();
+  int kernel_size = state.range(0);
+
+  for (auto _ : state) {
+    cv::Mat blurred;
+    cv::medianBlur(img, blurred, kernel_size);
+    benchmark::DoNotOptimize(blurred.data);
+  }
+}
+
+static void BM_Sobel(benchmark::State& state) {
+  cv::Mat img = GetImage();
+  int kernel_size = state.range(0);
+
+  for (auto _ : state) {
+    cv::Mat grad;
+    cv::Sobel(img, grad, CV_16S, 1, 0, kernel_size);
+    benchmark::DoNotOptimize(grad.data);
+  }
+}
+
+// Register kernel sizes for all filters
+BENCHMARK(BM_GaussianBlur)
+    ->Arg(3)
+    ->Arg(5)
+    ->Arg(7)
+    ->Arg(9)
+    ->Arg(11)
+    ->Arg(13)
+    ->Arg(15);
+BENCHMARK(BM_MedianBlur)
+    ->Arg(3)
+    ->Arg(5)
+    ->Arg(7)
+    ->Arg(9)
+    ->Arg(11)
+    ->Arg(13)
+    ->Arg(15);
+BENCHMARK(BM_Sobel)->Arg(3)->Arg(5)->Arg(7)->Arg(9)->Arg(11)->Arg(13)->Arg(15);

src/read_image.cpp

@@ -0,0 +1,11 @@
+#include <benchmark/benchmark.h>
+#include <opencv2/opencv.hpp>
+
+static void BM_ReadImage(benchmark::State& state) {
+  for (auto _ : state) {
+    cv::Mat img = cv::imread("assets/dog_bike_car.jpg");
+    benchmark::DoNotOptimize(img.data);
+    benchmark::ClobberMemory();
+  }
+}
+BENCHMARK(BM_ReadImage);

src/transform.cpp

@@ -0,0 +1,36 @@
+#include <benchmark/benchmark.h>
+#include <opencv2/opencv.hpp>
+
+static cv::Mat GetImage() {
+  static cv::Mat img = cv::imread("assets/dog_bike_car.jpg");
+  if (img.empty())
+    throw std::runtime_error("Cannot open image");
+  return img;
+}
+
+static void BM_Rotate(benchmark::State& state) {
+  cv::Mat img = GetImage();
+  double angle = state.range(0);
+  cv::Point2f center(img.cols / 2.0f, img.rows / 2.0f);
+  cv::Mat rot_mat = cv::getRotationMatrix2D(center, angle, 1.0);
+
+  for (auto _ : state) {
+    cv::Mat rotated;
+    cv::warpAffine(img, rotated, rot_mat, img.size());
+    benchmark::DoNotOptimize(rotated.data);
+  }
+}
+
+static void BM_Scale(benchmark::State& state) {
+  cv::Mat img = GetImage();
+  double scale = state.range(0) / 100.0;
+
+  for (auto _ : state) {
+    cv::Mat resized;
+    cv::resize(img, resized, {}, scale, scale);
+    benchmark::DoNotOptimize(resized.data);
+  }
+}
+
+BENCHMARK(BM_Rotate)->Arg(30)->Arg(45)->Arg(90);
+BENCHMARK(BM_Scale)->Arg(50)->Arg(200);