提取圖片中任意形狀的區域

openkk 12年前發布 | 33K 次閱讀 圖形/圖像處理 OpenCV

    正在做一個基于內容的圖像檢索項目,客戶要求可以讓用戶提取圖片上的任意形狀的子圖作為輸入。為了實現這個功能,花了不少精力,中間繞了不少圈子。現在問題總算解決了,把主要的思路記錄一下,希望能幫到有同樣需求的朋友。

      由于這個項目使用了第三方的開源圖像庫opencv,所以這個功能也是借助opencv來實現的。

      首先需要解決的是在圖片中繪制曲線的問題,思路很簡單,只需要響應鼠標事件通過描點、連線的方式就可以完成。在opencv中需要使用回調來響應鼠標事件,opencv中文站上有很好的示例,關鍵代碼:

#include "cv.h"

include "highgui.h"

include

<stdio.h>

include

<stdlib.h>







IplImage inpaint_mask = 0; IplImage img0 = 0, img = 0, inpainted = 0; CvPoint prev_pt = {-1,-1};

void on_mouse( int event, int x, int y, int flags, void* zhang) { if( !img ) return; if( event == CV_EVENT_LBUTTONUP || !(flags & CV_EVENT_FLAG_LBUTTON) ) prev_pt = cvPoint(-1,-1); else if( event == CV_EVENT_LBUTTONDOWN ) prev_pt = cvPoint(x,y); else if( event == CV_EVENT_MOUSEMOVE && (flags & CV_EVENT_FLAG_LBUTTON) ) { CvPoint pt = cvPoint(x,y); if( prev_pt.x < 0 ) prev_pt = pt; cvLine( inpaint_mask, prev_pt, pt, cvScalarAll(255), 5, 8, 0 ); cvLine( img, prev_pt, pt, cvScalarAll(255), 5, 8, 0 ); prev_pt = pt; cvShowImage( "image", img ); } }

int main( int argc, char* argv ) { char filename = argc >= 2 ? argv[1] : (char*)"fruits.jpg"; if( (img0 = cvLoadImage(filename,-1)) == 0 ) return 0; printf( "Hot keys: \n" cvNamedWindow( "image", 1 ); img = cvCloneImage( img0 ); inpainted = cvCloneImage( img0 ); inpaint_mask = cvCreateImage( cvGetSize(img), 8, 1 ); cvZero( inpaint_mask ); cvZero( inpainted ); cvShowImage( "image", img ); cvShowImage( "watershed transform", inpainted ); cvSetMouseCallback( "image", on_mouse, 0 );
}

</stdlib.h>





</stdio.h></pre>效果如下:

             提取圖片中任意形狀的區域

         第二步,利用封閉曲線提取圖形,時間和精力都主要浪費在這一塊了,最開始的想法是通過掃描線的方法獲取,但最終做出來的效果差強人意,連自己這關都過不了。最后幾經轉折,找到了這樣一個函數cvFloodFill(),這個函數可根據邊界填充圖像連通域,正是我所需要的。關鍵代碼:

        if(event == CV_EVENT_RBUTTONUP)
    {
        cvFloodFill(maskImg,cvPoint(x,y),cvScalarAll(255));
        cvSaveImage("maskImg.bmp",maskImg);
        IplImage *segImage=cvCreateImage(cvGetSize(img),8,3);;
        cvCopy(img,segImage,maskImg);
        cvSaveImage("segImage.bmp",segImage);
        cvCopy(segImage,proc->img);
        cvReleaseImage(&segImage);
        cvDestroyWindow("Key Image");
    }

 

效果如下:

            提取圖片中任意形狀的區域

 本文由用戶 openkk 自行上傳分享,僅供網友學習交流。所有權歸原作者,若您的權利被侵害,請聯系管理員。
 轉載本站原創文章,請注明出處,并保留原始鏈接、圖片水印。
 本站是一個以用戶分享為主的開源技術平臺,歡迎各類分享!