vendor: OpenCV 5.0.0 snapshot at 755e50675d97db9b7d449d8bd6b09888646f6c6e

modules/xphoto/samples/inpainting.cpp

@@ -0,0 +1,78 @@
+#include "opencv2/xphoto.hpp"
+
+#include "opencv2/imgproc.hpp"
+#include "opencv2/highgui.hpp"
+
+#include <ctime>
+#include <iostream>
+
+const char* keys =
+{
+    "{i || input image name}"
+    "{m || mask image name}"
+    "{o || output image name}"
+};
+
+int main( int argc, const char** argv )
+{
+    bool printHelp = ( argc == 1 );
+    printHelp = printHelp || ( argc == 2 && std::string(argv[1]) == "--help" );
+    printHelp = printHelp || ( argc == 2 && std::string(argv[1]) == "-h" );
+
+    if ( printHelp )
+    {
+        printf("\nThis sample demonstrates shift-map image inpainting\n"
+            "Call:\n"
+            "    inpainting -i=<string> -m=<string> [-o=<string>]\n\n");
+        return 0;
+    }
+
+    cv::CommandLineParser parser(argc, argv, keys);
+    if ( !parser.check() )
+    {
+        parser.printErrors();
+        return -1;
+    }
+
+    std::string inFilename = parser.get<std::string>("i");
+    std::string maskFilename = parser.get<std::string>("m");
+    std::string outFilename = parser.get<std::string>("o");
+
+    cv::Mat src = cv::imread(inFilename, cv::IMREAD_UNCHANGED);
+    if ( src.empty() )
+    {
+        printf( "Cannot read image file: %s\n", inFilename.c_str() );
+        return -1;
+    }
+
+    cv::cvtColor(src, src, cv::COLOR_BGR2Lab);
+
+    cv::Mat mask = cv::imread(maskFilename, cv::IMREAD_GRAYSCALE);
+    if ( mask.empty() )
+    {
+        printf( "Cannot read image file: %s\n", maskFilename.c_str() );
+        return -1;
+    }
+    cv::threshold(mask, mask, 128, 255, cv::THRESH_BINARY | cv::THRESH_OTSU);
+
+    cv::Mat res(src.size(), src.type());
+
+    int time = clock();
+    cv::xphoto::inpaint( src, mask, res, cv::xphoto::INPAINT_SHIFTMAP );
+    std::cout << "time = " << (clock() - time)
+        / double(CLOCKS_PER_SEC) << std::endl;
+
+    cv::cvtColor(res, res, cv::COLOR_Lab2BGR);
+
+    if ( outFilename == "" )
+    {
+        cv::namedWindow("inpainting result", 1);
+        cv::imshow("inpainting result", res);
+
+        cv::waitKey(0);
+    }
+    else
+        cv::imwrite(outFilename, res);
+
+    return 0;
+}