90 lines
2.2 KiB
C++
90 lines
2.2 KiB
C++
#include "opencv2/highgui/highgui.hpp"
|
|
#include "opencv2/imgproc/imgproc.hpp"
|
|
|
|
#include <iostream>
|
|
|
|
using namespace cv;
|
|
using namespace std;
|
|
|
|
void help()
|
|
{
|
|
cout << "\nCool inpainging demo. Inpainting repairs damage to images by floodfilling the damage \n"
|
|
<< "with surrounding image areas.\n"
|
|
"Using OpenCV version %s\n" << CV_VERSION << "\n"
|
|
"Usage:\n"
|
|
"./inpaint [image_name -- Default fruits.jpg]\n" << endl;
|
|
|
|
cout << "Hot keys: \n"
|
|
"\tESC - quit the program\n"
|
|
"\tr - restore the original image\n"
|
|
"\ti or SPACE - run inpainting algorithm\n"
|
|
"\t\t(before running it, paint something on the image)\n" << endl;
|
|
}
|
|
|
|
Mat img, inpaintMask;
|
|
Point prevPt(-1,-1);
|
|
|
|
void onMouse( int event, int x, int y, int flags, void* )
|
|
{
|
|
if( event == CV_EVENT_LBUTTONUP || !(flags & CV_EVENT_FLAG_LBUTTON) )
|
|
prevPt = Point(-1,-1);
|
|
else if( event == CV_EVENT_LBUTTONDOWN )
|
|
prevPt = Point(x,y);
|
|
else if( event == CV_EVENT_MOUSEMOVE && (flags & CV_EVENT_FLAG_LBUTTON) )
|
|
{
|
|
Point pt(x,y);
|
|
if( prevPt.x < 0 )
|
|
prevPt = pt;
|
|
line( inpaintMask, prevPt, pt, Scalar::all(255), 5, 8, 0 );
|
|
line( img, prevPt, pt, Scalar::all(255), 5, 8, 0 );
|
|
prevPt = pt;
|
|
imshow("image", img);
|
|
}
|
|
}
|
|
|
|
|
|
int main( int argc, char** argv )
|
|
{
|
|
char* filename = argc >= 2 ? argv[1] : (char*)"fruits.jpg";
|
|
Mat img0 = imread(filename, -1);
|
|
if(img0.empty())
|
|
{
|
|
cout << "Couldn't open the image " << filename << ". Usage: inpaint <image_name>\n" << endl;
|
|
return 0;
|
|
}
|
|
|
|
help();
|
|
|
|
namedWindow( "image", 1 );
|
|
|
|
img = img0.clone();
|
|
inpaintMask = Mat::zeros(img.size(), CV_8U);
|
|
|
|
imshow("image", img);
|
|
setMouseCallback( "image", onMouse, 0 );
|
|
|
|
for(;;)
|
|
{
|
|
char c = (char)waitKey();
|
|
|
|
if( c == 27 )
|
|
break;
|
|
|
|
if( c == 'r' )
|
|
{
|
|
inpaintMask = Scalar::all(0);
|
|
img0.copyTo(img);
|
|
imshow("image", img);
|
|
}
|
|
|
|
if( c == 'i' || c == ' ' )
|
|
{
|
|
Mat inpainted;
|
|
inpaint(img, inpaintMask, inpainted, 3, CV_INPAINT_TELEA);
|
|
imshow("inpainted image", inpainted);
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|