From ef2588ffcf5c2d440cc38f2406f8b59c1940ed50 Mon Sep 17 00:00:00 2001 From: saishock1504 Date: Sat, 24 Feb 2024 22:12:56 +0530 Subject: [PATCH] Updated erosion function to get correct result for different kernel sizes --- 4_cv_basics/6_morphology/src/morphology.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/4_cv_basics/6_morphology/src/morphology.cpp b/4_cv_basics/6_morphology/src/morphology.cpp index bd40c52..f48293a 100644 --- a/4_cv_basics/6_morphology/src/morphology.cpp +++ b/4_cv_basics/6_morphology/src/morphology.cpp @@ -27,7 +27,7 @@ int kernel_sum(Mat image, int row, int col, int Kernel_size) { if (Kernel_size % 2 != 1 || Kernel_size < 3) { - std::cout << "Kernel size should be of odd and grater than or equal to 3"; + std::cout << "Kernel size should be of odd and greater than or equal to 3"; return -1; } @@ -53,7 +53,7 @@ Mat erosion(Mat source_image, Mat output_image, int Kernel_size) { for (int j = 0; j < source_image.cols; j++) { - if (kernel_sum(source_image, i, j, Kernel_size) != 255 * 9) + if (kernel_sum(source_image, i, j, Kernel_size) != 255 * (Kernel_size*Kernel_size)) { output_image.at(i, j) = saturate_cast(0); }