test_canny.cpp

#include "my_cv/filters.h"
#include "my_cv/cv_commons.hpp"

#include "opencv2/highgui.hpp"
#include "opencv2/imgproc.hpp"

#include <vector>
#include <iostream>
#include <cmath>
#include <string>

void test_canny()
{

    // -- Read image
    const std::string filename = "data/color_chessboard.jpg";
    // const std::string filename = "data/simple_shapes2.png";
    cv::Mat src_gray;
    cv::cvtColor(cv_commons::readImage(filename), src_gray, cv::COLOR_BGR2GRAY);

    // -- Canny(Mine).
    const double lb = 800.0, ub = lb * 3.0;
    cv::Mat1b edge_canny = filters::canny(src_gray, lb, ub);

    // -- Canny(OpenCV).
    cv::Mat1b edge_canny_opencv;
    const int kernel_size = 5;
    cv::Canny(src_gray, edge_canny_opencv, lb, ub, kernel_size);

    // -- For comparison, compute sobel.
    constexpr bool TAKE_ABS = true;
    cv::Mat1b edge_sobel = cv_commons::float2uint8(filters::sobel(src_gray), TAKE_ABS);

    // -- Show image.
    const std::string WINDOW_NAME = "Original / Sobel / Canny(Mine) / Canny(OpenCV)";
    cv::Mat img_disp = cv_commons::display_images(
        {src_gray, edge_sobel, edge_canny, edge_canny_opencv},
        WINDOW_NAME);
    cv::imwrite("output/test_canny.png", img_disp);
}

int main(int argc, char const *argv[])
{
    test_canny();
}