亲宝软件园·资讯

展开

OpenCV选择图像中矩形区域并保存 OpenCV选择图像中矩形区域并保存

大唐游子 人气:0


根据《Learning OpenCV》中的example4.1改写:

// An example program in which the
// user can draw boxes on the screen.
//
//#include <cv.h>
//#include <highgui.h>
#include "opencv2/imgproc/imgproc.hpp"
#include "opencv2/highgui/highgui.hpp"
using namespace cv;
// Define our callback which we will install for
// mouse events.
//
void my_mouse_callback(
 int event, int x, int y, int flags, void* param
);
 
CvRect box;
bool drawing_box = false;
bool isRectDrawn = false;
 
// A litte subroutine to draw a box onto an image_copy
void draw_box( IplImage* img, CvRect rect ) {
 cvRectangle (
 img,
 cvPoint(box.x,box.y),
 cvPoint(box.x+box.width,box.y+box.height),
 cvScalar(0x00,0x00,0xff) /* blue */
 );
}
 
void draw_box_green( IplImage* img, CvRect rect ) {
 cvRectangle (
 img,
 cvPoint(box.x,box.y),
 cvPoint(box.x+box.width,box.y+box.height),
 cvScalar(0x00,0xff,0x00) /* green */
 );
}
 
int main( int argc, char* argv[] ) {
 
 box = cvRect(-1,-1,0,0);
 IplImage* image_input = cvLoadImage(argv[1]);
 IplImage* image = cvCloneImage( image_input );
 IplImage* image_copy = cvCloneImage( image );
 IplImage* temp = cvCloneImage( image_copy );
 cvNamedWindow( "Box Example" );
 // Here is the crucial moment that we actually install
 // the callback. Note that we set the value ‘param' to
 // be the image_copy we are working with so that the callback
 // will have the image_copy to edit.
 //
 cvSetMouseCallback(
 "Box Example",
 my_mouse_callback,
 (void*) image_copy
 );
 // The main program loop. Here we copy the working image_copy
 // to the ‘temp' image_copy, and if the user is drawing, then
 // put the currently contemplated box onto that temp image_copy.
 // display the temp image_copy, and wait 15ms for a keystroke,
 // then repeat…
 //
 while( 1 ) {
 //cvCopyImage( image_copy, temp );
 cvCopy( image_copy, temp );
 if( drawing_box ) draw_box( temp, box );
 cvShowImage( "Box Example", temp );
 //if( cvWaitKey( 15 )==27 ) break;
 int key = cvWaitKey( 15 );
 if(key == 27) break;
 if(isRectDrawn){
 if(key == 's' || key == 'S'){
 // draw green box
 draw_box_green( image_copy, box );
 cvCopy( image_copy, image );
 
 // save roi image
 static int index = 0;
 char save_image_name[128];
 sprintf(save_image_name, "rect_%d.jpg", index++);
 cvSetImageROI(image_input, box);
 cvSaveImage(save_image_name, image_input);
 cvResetImageROI(image_input);
 
 isRectDrawn = false;
 }
 
 if(key == 'q' || key == 'Q'){
 cvCopy( image, image_copy );
 isRectDrawn = false;
 }
 }
 }
 // Be tidy
 //
 cvReleaseImage( &image_copy );
 cvReleaseImage( &temp );
 cvDestroyWindow( "Box Example" );
}
 
// This is our mouse callback. If the user
// presses the left button, we start a box.
// when the user releases that button, then we
// add the box to the current image_copy. When the
// mouse is dragged (with the button down) we
// resize the box.
//
void my_mouse_callback(
int event, int x, int y, int flags, void* param
) {
 IplImage* image_copy = (IplImage*) param;
 switch( event ) {
 case CV_EVENT_MOUSEMOVE: {
 if( drawing_box ) {
 box.width = x-box.x;
 box.height = y-box.y;
 }
 }
 break;
 case CV_EVENT_LBUTTONDOWN: {
 drawing_box = true;
 box = cvRect(x, y, 0, 0);
 }
 break;
 case CV_EVENT_LBUTTONUP: {
 drawing_box = false;
 isRectDrawn = true;
 if(box.width<0) {
 box.x+=box.width;
 box.width *=-1;
 }
 if(box.height<0) {
 box.y+=box.height;
 box.height*=-1;
 }
 draw_box(image_copy, box);
 }
 break;
 }
}

使用方法:

载入图像后,用鼠标在图像上点击确定矩形起始点,拖动鼠标画矩形,抬起鼠标键时会画出一个红色矩形区域。按下s或S键,红色矩形变成绿色,并保存这个ROI区域。如果按下q或Q键,将会取消这次选择,红色矩形框消失。可以连续选取多个区域。按ESC键退出程序。

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持。

加载全部内容

相关教程
猜你喜欢
用户评论