OpenCV 实践程序17——实现图片标注

该程序实验在一文件夹下自动加载jpg格式（格式可以改）的图片，我们用bouningbox选取我们感兴趣的物体，程序会自动记录最小点坐标和bouningbox的长和宽，并保存到info文本文本中。

#include "cv.h"
#include "cvaux.h"
#include "highgui.h"

// for filelisting
#include <stdio.h>
#include <io.h>
// for fileoutput
#include <string>
#include <fstream>
#include <sstream>

using namespace std;

IplImage* image=0;
IplImage* image2=0;
int start_roi=0;
int roi_x0=0;
int roi_y0=0;
int roi_x1=0;
int roi_y1=0;
int numOfRec=0;
char* window_name="<SPACE>add <ENTER>save and load next <ESC>exit";

string IntToString(int num)
{
	ostringstream myStream; //creates an ostringstream object
	myStream << num << flush;
	
	// outputs the number into the string stream and then flushes
	//the buffer (makes sure the output is put into the stream)
	
	return(myStream.str()); //returns the string form of the stringstream object
};

void onmouse(int event,int x,int y,int flag,void *param)
{
    if(event==CV_EVENT_LBUTTONDOWN)
    {
		start_roi=1;
		roi_x0=x;
		roi_y0=y;
    }
    if(event==CV_EVENT_MOUSEMOVE && flag==CV_EVENT_FLAG_LBUTTON)
    {
		roi_x1=x;
		roi_y1=y;

		//redraw ROI selection
		image2=cvCloneImage(image);
		cvRectangle(image2,cvPoint(roi_x0,roi_y0),cvPoint(roi_x1,roi_y1),CV_RGB(255,0,255),1);
		cvShowImage(window_name,image2);
		cvReleaseImage(&image2);
    }
    if(event==CV_EVENT_LBUTTONUP)
    {
		start_roi=0;
    }
}

int main(int argc, char** argv)
{
	struct _finddata_t bmp_file;
    long hFile;
	int iKey=0;
	
//	get *.bmp files in directory
	if((hFile=_findfirst("rawdata/*.jpg",&bmp_file))==-1L)
		printf("no *.jpg files in directory 'rawdata'\n");
	else
	{

	//	init highgui
		cvAddSearchPath("rawdata");
		cvNamedWindow(window_name,1);
		cvSetMouseCallback(window_name,onmouse);

	//	init output of rectangles to the info file
		ofstream output("info.txt");

		string strPrefix;
		string strPostfix;

	//	open every *.bmp file
		do
		{
			printf(" %-12s\n",bmp_file.name);
			
			numOfRec=0;
			strPostfix="";
			strPrefix="rawdata/";
			strPrefix+=bmp_file.name;
	//		bmp_file.name=strPrefix+bmp_file.name;
			
	//		image=cvLoadImage(bmp_file.name,1);
			image=cvLoadImage(strPrefix.c_str(),1);
		//	work on current image
			do
			{
				cvShowImage(window_name,image);
			
				// used cvWaitKey returns:
				//	<Enter>=13		save added rectangles and show next image
				//	<ESC>=27		exit program
				//	<Space>=32		add rectangle to current image
				//  any other key clears rectangle drawing only
				iKey=cvWaitKey(0);
				switch(iKey)
				{
				case 27:
						cvReleaseImage(&image);
						cvDestroyWindow(window_name);
						return 0;
				case 32:
						numOfRec++;
						// currently two draw directions possible:
						//		from top left to bottom right or vice versa
						if(roi_x0<roi_x1 && roi_y0<roi_y1)
						{
							//打印最小点坐标和长和宽。
							printf("   %d. rect x=%d\ty=%d\twidth=%d\theight=%d\n",numOfRec,roi_x0,roi_y0,roi_x1-roi_x0,roi_y1-roi_y0);

							// append rectangle coord to previous line content
							strPostfix+=" "+IntToString(roi_x0)+" "+IntToString(roi_y0)+" "+IntToString(roi_x1-roi_x0)+" "+IntToString(roi_y1-roi_y0);
						}
						if(roi_x0>roi_x1 && roi_y0>roi_y1)
						{
							printf("   %d. rect x=%d\ty=%d\twidth=%d\theight=%d\n",numOfRec,roi_x1,roi_y1,roi_x0-roi_x1,roi_y0-roi_y1);
							// append rectangle coord to previous line content
							strPostfix+=" "+IntToString(roi_x1)+" "+IntToString(roi_y1)+" "+IntToString(roi_x0-roi_x1)+" "+IntToString(roi_y0-roi_y1);
						}
						break;
				}
			}
			while(iKey!=13);
			
			// save to info file as later used for HaarTraining:
			//	<rel_path>\bmp_file.name numOfRec x0 y0 width0 height0 x1 y1 width1 height1...
			if(numOfRec>0 && iKey==13)
			{
				//append line
				output << strPrefix << " "<< numOfRec << strPostfix <<"\n";
			}

			cvReleaseImage(&image);
		}
		while(_findnext(hFile,&bmp_file)==0);
		
		output.close();
		cvDestroyWindow(window_name);
		_findclose( hFile );
	}

	system("pause");
	return 0;
}

这个程序实现记录归一化之后的中心坐标及长宽。

#include "cv.h"
#include "cvaux.h"
#include "highgui.h"

// for filelisting
#include <stdio.h>
#include <io.h>
// for fileoutput
#include <string>
#include <fstream>
#include <sstream>


using namespace std;

IplImage* image=0;
IplImage* image2=0;
int start_roi=0;
float roi_x0=0;
float roi_y0=0;
float roi_x1=0;
float roi_y1=0;
int numOfRec=0;
char* window_name="<SPACE>add <ENTER>save and load next <ESC>exit";

string IntToString(int num)
{
	ostringstream myStream; //creates an ostringstream object
	myStream << num << flush;
	
	// outputs the number into the string stream and then flushes
	//the buffer (makes sure the output is put into the stream)
	
	return(myStream.str()); //returns the string form of the stringstream object
};

void onmouse(int event,int x,int y,int flag,void *param)
{
    if(event==CV_EVENT_LBUTTONDOWN)
    {
		start_roi=1;
		roi_x0=x;
		roi_y0=y;
    }
    if(event==CV_EVENT_MOUSEMOVE && flag==CV_EVENT_FLAG_LBUTTON)
    {
		roi_x1=x;
		roi_y1=y;

		//redraw ROI selection
		image2=cvCloneImage(image);
		cvRectangle(image2,cvPoint(roi_x0,roi_y0),cvPoint(roi_x1,roi_y1),CV_RGB(255,0,255),1);
		cvShowImage(window_name,image2);
		cvReleaseImage(&image2);
    }
    if(event==CV_EVENT_LBUTTONUP)
    {
		start_roi=0;
    }
}

int main(int argc, char** argv)
{
	struct _finddata_t jpg_file;
    long hFile;
	int iKey=0;
	
//	get *.jpg files in directory
	if((hFile=_findfirst("train/*.jpg",&jpg_file))==-1L)
		printf("no *.jpg files in directory 'train'\n");
	else
	{

	//	init highgui
		cvAddSearchPath("train");
		cvNamedWindow(window_name,1);
		cvSetMouseCallback(window_name,onmouse);

	//	init output of rectangles to the info file
		ofstream output("info.txt");

		string strPrefix;
		string strPostfix;

	//	open every *.jpg file
		do
		{
			printf(" %-12s\n",jpg_file.name);
			
			numOfRec=0;
			strPostfix="";
			strPrefix="train/";
			strPrefix+=jpg_file.name;
	//		jpg_file.name=strPrefix+jpg_file.name;
			
	//		image=cvLoadImage(jpg_file.name,1);
			image=cvLoadImage(strPrefix.c_str(),1);
			
		//	work on current image
			do
			{
				cvShowImage(window_name,image);
			
				// used cvWaitKey returns:
				//	<Enter>=13		save added rectangles and show next image
				//	<ESC>=27		exit program
				//	<Space>=32		add rectangle to current image
				//  any other key clears rectangle drawing only
				iKey=cvWaitKey(0);
				switch(iKey)
				{
				case 27:
						cvReleaseImage(&image);
						cvDestroyWindow(window_name);
						return 0;
				case 32:
						numOfRec++;
						// currently two draw directions possible:
						//		from top left to bottom right or vice versa

						ostringstream buffer1,buffer2,buffer3,buffer4;

						if(roi_x0<roi_x1 && roi_y0<roi_y1)
						{
							
							float a=(roi_x0+roi_x1)/2.0;
							float b=(roi_y0+roi_y1)/2.0;
							//打印最小点坐标和长和宽。
							//printf("   %d. rect x=%d\ty=%d\twidth=%d\theight=%d\n",numOfRec,roi_x0,roi_y0,roi_x1-roi_x0,roi_y1-roi_y0);
							printf("   %d. rect x=%0.6f\ty=%0.6f\twidth=%0.6f\theight=%0.6f\n",numOfRec,a/image->width,b/image->height,(roi_x1-roi_x0)/image->width,(roi_y1-roi_y0)/image->height);

							buffer1 << a/image->width;buffer2 << b/image->height;buffer3<<(roi_x1-roi_x0)/image->width;buffer4<<(roi_y1-roi_y0)/image->height;

							// append rectangle coord to previous line content
							strPostfix+=" "+buffer1.str()+" "+buffer2.str()+" "+buffer3.str()+" "+buffer4.str();
						}
						if(roi_x0>roi_x1 && roi_y0>roi_y1)
						{
							float a=(roi_x0+roi_x1)/2.0;
							float b=(roi_y0+roi_y1)/2.0;
							printf("   %d. rect x=%0.6f\ty=%0.6f\twidth=%0.6f\theight=%0.6f\n",numOfRec,a/image->width,b/image->height,(roi_x0-roi_x1)/image->width,(roi_y0-roi_y1)/image->height);

							buffer1 << a/image->width;buffer2 << b/image->height;buffer3<<(roi_x0-roi_x1)/image->width;buffer4<<(roi_y0-roi_y1)/image->height;

							// append rectangle coord to previous line content
							strPostfix+=" "+buffer1.str()+" "+buffer2.str()+" "+buffer3.str()+" "+buffer4.str();
						}
						break;
				}
			}
			while(iKey!=13);
			
			// save to info file as later used for HaarTraining:
			//	<rel_path>\jpg_file.name numOfRec x0 y0 width0 height0 x1 y1 width1 height1...
			if(numOfRec>0 && iKey==13)
			{
				//append line
				output << strPrefix << " "<< numOfRec << strPostfix <<"\n";
			}

			cvReleaseImage(&image);
		}
		while(_findnext(hFile,&jpg_file)==0);
		
		output.close();
		cvDestroyWindow(window_name);
		_findclose( hFile );
	}

	system("pause");
	return 0;
}