From 82ca9e11965bce800ed5d500ab9c7dbc9ac3aa77 Mon Sep 17 00:00:00 2001 From: crupest Date: Thu, 2 Jun 2022 22:42:29 +0800 Subject: import(life): ... --- .../life/digital-image-process-lab/CMakeLists.txt | 3 ++ .../HarrisCornerDetector.cpp | 49 ++++++++++++++++++++++ 2 files changed, 52 insertions(+) create mode 100644 works/life/digital-image-process-lab/HarrisCornerDetector.cpp (limited to 'works/life') diff --git a/works/life/digital-image-process-lab/CMakeLists.txt b/works/life/digital-image-process-lab/CMakeLists.txt index 5c2d05b..8a00ca5 100644 --- a/works/life/digital-image-process-lab/CMakeLists.txt +++ b/works/life/digital-image-process-lab/CMakeLists.txt @@ -15,3 +15,6 @@ target_link_libraries(DiscreteFourierTransform ${OpenCV_LIBS}) add_executable(MorphologicalTransformation MorphologicalTransformation.cpp) target_link_libraries(MorphologicalTransformation ${OpenCV_LIBS}) + +add_executable(HarrisCornerDetector HarrisCornerDetector.cpp) +target_link_libraries(HarrisCornerDetector ${OpenCV_LIBS}) diff --git a/works/life/digital-image-process-lab/HarrisCornerDetector.cpp b/works/life/digital-image-process-lab/HarrisCornerDetector.cpp new file mode 100644 index 0000000..c7013be --- /dev/null +++ b/works/life/digital-image-process-lab/HarrisCornerDetector.cpp @@ -0,0 +1,49 @@ +#include +#include + +#include + +cv::Mat src, src_gray; +int thresh = 100; +int max_thresh = 255; +const char *source_window = "Source image"; +const char *corners_window = "Corners detected"; +void cornerHarris_demo(int, void *); +int main(int argc, char **argv) { + cv::CommandLineParser parser(argc, argv, + "{@input | building.jpg | input image}"); + src = cv::imread(cv::samples::findFile(parser.get("@input"))); + if (src.empty()) { + std::cout << "Could not open or find the image!\n" << std::endl; + std::cout << "Usage: " << argv[0] << " " << std::endl; + return -1; + } + cvtColor(src, src_gray, cv::COLOR_BGR2GRAY); + cv::namedWindow(source_window); + cv::createTrackbar("Threshold: ", source_window, &thresh, max_thresh, + cornerHarris_demo); + cv::imshow(source_window, src); + cornerHarris_demo(0, 0); + cv::waitKey(); + return 0; +} + +void cornerHarris_demo(int, void *) { + int blockSize = 2; + int apertureSize = 3; + double k = 0.04; + cv::Mat dst = cv::Mat::zeros(src.size(), CV_32FC1); + cv::cornerHarris(src_gray, dst, blockSize, apertureSize, k); + cv::Mat dst_norm, dst_norm_scaled; + normalize(dst, dst_norm, 0, 255, cv::NORM_MINMAX, CV_32FC1, cv::Mat()); + convertScaleAbs(dst_norm, dst_norm_scaled); + for (int i = 0; i < dst_norm.rows; i++) { + for (int j = 0; j < dst_norm.cols; j++) { + if ((int)dst_norm.at(i, j) > thresh) { + circle(dst_norm_scaled, cv::Point(j, i), 5, cv::Scalar(0), 2, 8, 0); + } + } + } + cv::namedWindow(corners_window); + cv::imshow(corners_window, dst_norm_scaled); +} -- cgit v1.2.3