70 lines
1.6 KiB
C++
70 lines
1.6 KiB
C++
|
#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);
|
||
|
|
|
||
|
|
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;
|
||
|
|
}
|