testing-repository/OpenCV_test/test10_preprocess/main.cpp

71 lines
1.6 KiB
C++
Raw Normal View History

2022-07-28 13:53:04 +08:00
#include <opencv2/opencv.hpp>
#include <iostream>
#include <fstream>
using namespace cv;
int main(int argc, char **argv)
{
std::string filename;
Mat image, gray_image, thre_image, canny_image;
int brightless = 0;
// Get Filename
std::cout << "Input Filename:";
std::cin >> filename;
// Load the image
if (filename != "")
{
image = imread(filename, IMREAD_COLOR);
if (image.empty())
{
std::cout << "Failed to open the image!" << std::endl;
return -1;
}
}
// Convert the image to HSV color and filter the background color
cvtColor(image, gray_image, COLOR_BGR2GRAY);
namedWindow("Original Image", WINDOW_AUTOSIZE);
imshow("Original Image", image);
namedWindow("Gray Image", WINDOW_AUTOSIZE);
imshow("Gray Image", gray_image);
// Use cv::thershold to convert
threshold(gray_image, thre_image, 0, 255, THRESH_BINARY | THRESH_OTSU);
namedWindow("OTSU Threshold", WINDOW_AUTOSIZE);
imwrite("/dev/shm/thre_test.png", thre_image);
imshow("OTSU Threshold", thre_image);
// Canny Process
Canny(thre_image, canny_image, 80, 150);
namedWindow("Canny Image", WINDOW_AUTOSIZE);
imshow("Canny Image", canny_image);
2022-07-28 15:48:11 +08:00
2022-07-28 13:53:04 +08:00
std::fstream outfile;
outfile.open("test.txt", std::ios_base::out);
if (outfile.is_open())
{
outfile << canny_image;
}
outfile.close();
while (1)
{
int c = waitKey(1);
if (c != -1)
{
std::cout << c << std::endl;
}
if (c == 27)
{
break;
}
}
destroyAllWindows();
return 0;
}