mirror of https://github.com/opencv/opencv.git
Open Source Computer Vision Library
https://opencv.org/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
91 lines
2.3 KiB
91 lines
2.3 KiB
#include "opencv2/imgcodecs.hpp" |
|
#include "opencv2/highgui/highgui.hpp" |
|
#include "opencv2/imgproc/imgproc.hpp" |
|
#include "opencv2/photo/photo.hpp" |
|
|
|
#include <iostream> |
|
|
|
using namespace cv; |
|
using namespace std; |
|
|
|
static 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 ../data/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); |
|
|
|
static void onMouse( int event, int x, int y, int flags, void* ) |
|
{ |
|
if( event == EVENT_LBUTTONUP || !(flags & EVENT_FLAG_LBUTTON) ) |
|
prevPt = Point(-1,-1); |
|
else if( event == EVENT_LBUTTONDOWN ) |
|
prevPt = Point(x,y); |
|
else if( event == EVENT_MOUSEMOVE && (flags & 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*)"../data/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, INPAINT_TELEA); |
|
imshow("inpainted image", inpainted); |
|
} |
|
} |
|
|
|
return 0; |
|
}
|
|
|