#include "stdafx.h"
#include "cv.h"
#include "camerads.h"
#include "highgui.h"
#include <stdio.h>
#include <ctype.h>
//#include <mmsystem.h>
IplImage* inpaint_mask = 0;
IplImage* img0 = 0, *img = 0, *inpainted = 0;
CvPoint prev_pt = {-1,-1};
IplImage *image = 0, *hsv = 0, *hue = 0, *mask = 0, *backproject = 0, *histimg = 0;
//IplImage *greyImage =0,*greyImage1 =0;
//int Thresholdness = 141;
//int ialpha = 20;
//int ibeta=20;
//int igamma=20;
CvHistogram *hist = 0;
int backproject_mode = 0;
int select_object = 0;
int track_object = 0;
int show_hist = 1;
int lv=2;
CvPoint origin;
CvPoint mCenter;
CvPoint mOffset[3];
POINT lpPoint;
CvRect selection;
CvRect track_window;
CvBox2D track_box;
CvConnectedComp track_comp;
int hdims = 16;
float hranges_arr[] = {0,180};
float* hranges = hranges_arr;
int vmin = 90, vmax = 256, smin = 90;
int frame_count=0;
CvScalar hsv2rgb( float hue );
//载入识别模板
void loadTemplateImage()
{
IplImage *tempimage = cvLoadImage("D:\\代码\\opencv\\hand detection\\Debug\\green.bmp",-1);
cvCvtColor( tempimage, hsv, CV_BGR2HSV );
int _vmin = vmin, _vmax = vmax;
cvInRangeS( hsv, cvScalar(0,smin,MIN(_vmin,_vmax),0),
cvScalar(180,256,MAX(_vmin,_vmax),0), mask );
cvSplit( hsv, hue, 0, 0, 0 );
selection.x = 1;
selection.y = 1;
selection.width = 320-1;
selection.height= 240-1;
mCenter.x=-1;
mCenter.y=-1;
cvSetImageROI( hue, selection );
cvSetImageROI( mask, selection );
cvCalcHist( &hue, hist, 0, mask );
float max_val = 0.f;
cvGetMinMaxHistValue( hist, 0, &max_val, 0, 0 );
cvConvertScale( hist->bins, hist->bins, max_val ? 255. / max_val : 0., 0 );
cvResetImageROI( hue );
cvResetImageROI( mask );
track_window = selection;
track_object = 1;
cvZero( histimg );
int bin_w = histimg->width / hdims;
for(int i = 0; i < hdims; i++ )
{
int val = cvRound( cvGetReal1D(hist->bins,i)*histimg->height/255 );
CvScalar color = hsv2rgb(i*180.f/hdims);
cvRectangle( histimg, cvPoint(i*bin_w,histimg->height),
cvPoint((i+1)*bin_w,histimg->height - val),
color, -1, 8, 0 );
}
cvReleaseImage(&tempimage);
}
//使用direct show打开摄像头
int open_Camera(CCameraDS *camera)
{
int cam_count;
//获取摄像头数目
cam_count = CCameraDS::CameraCount();
printf("There are %d cameras.\n", cam_count);
if(cam_count==0)
return -1;
if(! camera->OpenCamera(0))
{
fprintf(stderr, "Can not open camera.\n");
return -1;
}
return cam_count;
}
//使用鼠标绘图
void on_mouse( int event, int x, int y, int flags, void* )
{
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( "Histogram", img );
}
}
CvScalar hsv2rgb( float hue )
{
int rgb[3], p, sector;
static const int sector_data[][3]=
{{0,2,1}, {1,2,0}, {1,0,2}, {2,0,1}, {2,1,0}, {0,1,2}};
hue *= 0.033333333333333333333333333333333f;
sector = cvFloor(hue);
p = cvRound(255*(hue - sector));
p ^= sector & 1 ? 255 : 0;
rgb[sector_data[sector][0]] = 255;
rgb[sector_data[sector][1]] = 0;
rgb[sector_data[sector][2]] = p;
return cvScalar(rgb[2], rgb[1], rgb[0],0);
}
int main( int argc, char** argv )
{
char* filename = argc >= 2 ? argv[1] : (char*)"fruits.jpg";
if( (img0 = cvLoadImage(filename,-1)) == 0 )
return 0;
int cam_count=0;
CCameraDS camera;
if((cam_count=open_Camera(&camera))<1){
printf("no camera find!\n");
return 0;
}
printf( "Hot keys: \n"
"\tESC - quit the program\n"
"\tb - switch to/from backprojection view\n"
"\th - show/hide object histogram\n"
"\tr - restore the original image\n"
"\ti or ENTER - run inpainting algorithm\n"
"\t\t(before running it, paint something on the image"
"To initialize tracking, select the object with mouse\n" );
// HWND hWnd=GetDesktopWindow();
// const char* desktop=cvGetWindowName( hWnd);
// printf("desk=%s\n",desktop);
cvNamedWindow( "Histogram", 1 );
cvNamedWindow( "CamShiftDemo", 1 );
cvSetMouseCallback( "CamShiftDemo", on_mouse, 0 );
cvCreateTrackbar( "Vmin", "CamShiftDemo", &vmin, 256, 0 );
cvCreateTrackbar( "Vmax", "CamShiftDemo", &vmax, 256, 0 );
cvCreateTrackbar( "Smin", "CamShiftDemo", &smin, 256, 0 );
img = cvCloneImage( img0 );
inpainted = cvCloneImage( img0 );
inpaint_mask = cvCreateImage( cvGetSize(img), 8, 1 );
cvZero( inpaint_mask );
cvZero( inpainted );
cvShowImage( "Histogram", img );
cvShowImage( "watershed transform", inpainted );
cvSetMouseCallback( "Histogram", on_mouse, 0 );
for(int i=0;i<3;i++)
{
mOffset[i].x=0;
mOffset[i].y=0;
}
for(;;)
{
IplImage* frame = 0;
int i, bin_w, c;
frame=camera.QueryFrame();
// frame = cvQueryFrame( capture );
if( !frame )
break;
if( !image)
{
// allocate all the buffers
image = cvCreateImage( cvGetSize(frame), 8, 3 );
image->origin = frame->origin;
hsv = cvCreateImage( cvGetSize(frame), 8, 3 );
hue = cvCreateImage( cvGetSize(frame), 8, 1 );
mask = cvCreateImage( cvGetSize(frame), 8, 1 );
backproject = cvCreateImage( cvGetSize(frame), 8, 1 );
hist = cvCreateHist( 1, &hdims, CV_HIST_ARRAY, &hranges, 1 );
histimg = cvCreateImage( cvGetSize(frame), 8, 3 );
cvZero( histimg );
loadTemplateImage();
}
cvCopy( frame, image, 0 );
cvCvtColor( image, hsv, CV_BGR2HSV );
if( track_object )
{
frame_count++;
frame_count=frame_count%3;
int _vmin = vmin, _vmax = vmax;
cvInRangeS( hsv, cvScalar(0,smin,MIN(_vmin,_vmax),0),
cvScalar(180,256,MAX(_vmin,_vmax),0), mask );
cvSplit( hsv, hue, 0, 0, 0 );
cvCalcBackProject( &hue, backproject, hist );
cvAnd( backproject, mask, backproject, 0 );
cvCamShift( backproject, track_window,
cvTermCriteria( CV_TERMCRIT_EPS | CV_TERMCRIT_ITER, 10, 1 ),
&track_comp, &track_box );
track_window = track_comp.rect;
if( backproject_mode )
cvCvtColor( backproject, image, CV_GRAY2BGR );
if( image->origin )
track_box.angle = -track_box.angle;
//中心
if(mCenter.x!=-1){
mOffset[frame_count].x= lv*(int) (track_box.center.x-mCenter.x);
mOffset[frame_count].y= lv*(int) (track_box.center.y-mCenter.y);
}
mCenter.x = (int) track_box.center.x;
mCenter.y = (int) track_box.center.y;
int offsetx=int((mOffset[0].x+mOffset[1].x+mOffset[2].x)/3);
int offsety=int((mOffset[0].y+mOffset[1].y+mOffset[2].y)/3);
GetCursorPos(&lpPoint);
SetCursorPos(lpPoint.x+offsetx,lpPoint.y+offsety);
printf("cx=%d,cy=%d,sizex=%02f,sizey=%02f\n",mCenter.x,mCenter.y,
track_box.size.width,track_box.size.height);
if(mCenter.x<0||mCenter.y<0||track_box.size.width<0||track_box.size.height<0)
{continue;}
cvEllipseBox( image, track_box, CV_RGB(255,0,0), 3, CV_AA, 0 );
}
if( select_object && selection.width > 0 && selection.height > 0 )
- 1
- 2
前往页