原始圖像

這段代碼展示了一個(gè)使用OpenCV庫進(jìn)行圖像修復(fù)的例子。它首先包含了處理圖像編碼、解碼、顯示、處理和照片處理所必要的OpenCV模塊的頭文件。然后利用cv和std命名空間下的類和方法。通過定義一個(gè)鼠標(biāo)回調(diào)函數(shù)onMouse來處理圖像上的繪圖操作，并通過主函數(shù)main處理圖像讀取，修復(fù)與顯示。

整體功能是：程序允許用戶用鼠標(biāo)在圖像上繪制區(qū)域，然后用指定的修復(fù)算法（圖像修復(fù)算法）來修復(fù)這些區(qū)域。用戶可以通過按鍵來選擇恢復(fù)原始圖像或者運(yùn)行修復(fù)算法，最后顯示修復(fù)后的結(jié)果。

#include "opencv2/imgcodecs.hpp" // 包含OpenCV模塊中處理圖像編碼和解碼的頭文件
#include "opencv2/highgui.hpp" // 包含OpenCV模塊中用于創(chuàng)建界面以顯示和保存圖像和視頻的頭文件
#include "opencv2/imgproc.hpp" // 包含OpenCV模塊中處理圖像處理的頭文件
#include "opencv2/photo.hpp" // 包含OpenCV模塊中處理照片的頭文件


#include <iostream> // 包含標(biāo)準(zhǔn)輸入輸出流的頭文件


using namespace cv; // 使用命名空間cv，這樣就可以直接使用OpenCV中定義的類和方法，而不需要加cv::前綴
using namespace std; // 使用命名空間std，便于使用標(biāo)準(zhǔn)庫中的功能


static void help( char** argv ) // 定義幫助函數(shù)，打印程序使用方法
{
    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" << argv[0] <<" [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; // 聲明Mat類變量img和inpaintMask，分別用于存儲(chǔ)圖像和繪制修復(fù)區(qū)域的掩碼
Point prevPt(-1,-1); // 聲明Point類變量prevPt，并初始化為(-1,-1)，用于存儲(chǔ)上一個(gè)繪圖點(diǎn)的位置


static void onMouse( int event, int x, int y, int flags, void* ) // 定義鼠標(biāo)回調(diào)函數(shù)，用于處理鼠標(biāo)事件
{
    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 ) // 主函數(shù)入口
{
    cv::CommandLineParser parser(argc, argv, "{@image|fruits.jpg|}"); // 解析命令行參數(shù)
    help(argv); // 調(diào)用幫助函數(shù)顯示使用方法


    string filename = samples::findFile(parser.get<string>("@image")); // 獲取圖像文件的路徑
    Mat img0 = imread(filename, IMREAD_COLOR); // 讀取圖像文件
    if(img0.empty()) // 如果圖像為空，則顯示錯(cuò)誤信息并返回
    {
        cout << "Couldn't open the image " << filename << ". Usage: inpaint <image_name>\n" << endl;
        return 0;
    }


    namedWindow("image", WINDOW_AUTOSIZE); // 創(chuàng)建顯示圖像的窗口


    img = img0.clone(); // 克隆原始圖像
    inpaintMask = Mat::zeros(img.size(), CV_8U); // 創(chuàng)建掩碼圖像，并初始化所有像素為0


    imshow("image", img); // 顯示原始圖像
    setMouseCallback( "image", onMouse, NULL); // 設(shè)置鼠標(biāo)回調(diào)函數(shù)


    for(;;) // 無限循環(huán)，等待用戶輸入
    {
        char c = (char)waitKey(); // 等待鍵盤輸入


        if( c == 27 ) // 如果按下ESC鍵，則退出程序
            break;


        if( c == 'r' ) // 如果按下'r'鍵，則恢復(fù)原始圖像
        {
            inpaintMask = Scalar::all(0);
            img0.copyTo(img);
            imshow("image", img);
        }


        if( c == 'i' || c == ' ' ) // 如果按下'i'鍵或空格鍵，則運(yùn)行修復(fù)算法
        {
            Mat inpainted;
            inpaint(img, inpaintMask, inpainted, 3, INPAINT_TELEA); // 運(yùn)行修復(fù)算法
            imshow("inpainted image", inpainted); // 顯示修復(fù)后的圖像
        }
    }


    return 0; // 程序結(jié)束
}

inpaint(img, inpaintMask, inpainted, 3, INPAINT_TELEA);

文章來源地址http://www.zghlxwxcb.cn/news/detail-857099.html