OpenCV_提取圖片中任意形狀的區域
轉載自:http://blog.csdn.net/hanbinga17/article/details/6536331#
----------------------------------------------------------------------------------------------------------------------------------------------
正在做一個基於內容的圖像檢索項目,客戶要求可以讓用戶提取圖片上的任意形狀的子圖作爲輸入。爲了實現這個功能,花了不少精力,中間繞了不少圈子。現在問題總算解決了,把主要的思路記錄一下,希望能幫到有同樣需求的朋友。
由於這個項目使用了第三方的開源圖像庫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 );
- }
效果如下:
第二步,利用封閉曲線提取圖形,時間和精力都主要浪費在這一塊了,最開始的想法是通過掃描線的方法獲取,但最終做出來的效果差強人意,連自己這關都過不了。最後幾經轉折,找到了這樣一個函數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");
- }
效果如下: