#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; }