vehicle detection haarcascades

Vehicle Detection Haarcascades

Github: https://github.com/andrewssobral/vehicle_detection_haarcascades

源码

#include <stdio.h>
#include <cv.h>
#include <highgui.h>

CvHaarClassifierCascade *cascade;
CvMemStorage            *storage;

void detect(IplImage *img);

int main(int argc, char** argv)
{
CvCapture *capture;
IplImage  *frame;
int input_resize_percent = 100;

cascade = (CvHaarClassifierCascade*)cvLoad("cars3.xml", 0, 0, 0);
storage = cvCreateMemStorage(0);
capture = cvCaptureFromAVI("video1.avi");

assert(cascade && storage && capture);

cvNamedWindow("video", 1);

const int KEY_SPACE = 32;
const int KEY_ESC = 27;

int key = 0;
do
{
frame = cvQueryFrame(capture);

if (!frame)
break;

//   cvResize(frame1, frame);

detect(frame);

key = cvWaitKey(10);

if (key == KEY_SPACE)
key = cvWaitKey(0);

if (key == KEY_ESC)
break;

} while (1);

cvDestroyAllWindows();
cvReleaseImage(&frame);
cvReleaseCapture(&capture);
cvReleaseHaarClassifierCascade(&cascade);
cvReleaseMemStorage(&storage);

return 0;
}

void detect(IplImage *img)
{
CvSize img_size = cvGetSize(img);
CvSeq *object = cvHaarDetectObjects(
img,
cascade,
storage,
1.1, //1.1,//1.5, //-------------------SCALE FACTOR
1, //2        //------------------MIN NEIGHBOURS
0, //CV_HAAR_DO_CANNY_PRUNING
cvSize(0, 0),//cvSize( 30,30), // ------MINSIZE
img_size //cvSize(70,70)//cvSize(640,480)  //---------MAXSIZE
);

std::cout << "Total: " << object->total << " cars" << std::endl;
for (int i = 0; i < (object ? object->total : 0); i++)
{
CvRect *r = (CvRect*)cvGetSeqElem(object, i);
cvRectangle(img,
cvPoint(r->x, r->y),
cvPoint(r->x + r->width, r->y + r->height),
CV_RGB(255, 0, 0), 2, 8, 0);
}

cvShowImage("video", img);
}

检测效果

车辆检测其他相关博客