简体   繁体   English

OpenCV仅在大轮廓周围绘制矩形?

[英]OpenCV draw rectangles around only large contours?

first time posting, hope I put the code in the right way. 第一次发布时,希望我以正确的方式输入代码。

I'm trying to detect and count vehicles in a video, and so if you look at my code below I find the contours of the image after thresholding and dilating, and then I use drawContours and rectangle to draw a box around the detected contours. 我正在尝试检测和计数视频中的车辆,因此,如果您看下面的代码,在阈值化和扩张之后,我会找到图像的轮廓,然后使用drawContours和矩形在检测到的轮廓周围绘制一个框。

I tried to put a filter on the drawContours/rectangle if statement by saying if the area of the rectangle isn't bigger than 40,000, then don't draw it. 我试图通过说矩形的面积是否不大于40,000来在drawContours / rectangle if语句上放置一个过滤器,所以不要绘制它。

Now if you look at the picture I attached, you'll see that there are rectangles that are drawn inside the larger rectangles, which I am not trying to do. 现在,如果您看一下我附带的图片,您会看到在较大的矩形内绘制了一些矩形,而我并不想这样做。 enter image description here . 在此处输入图片说明 These rectangles has an area less than 40,000 but they are being drawn for some reason. 这些矩形的面积小于40,000,但由于某些原因而被绘制。

I was going to use the rectangles to count the cars on the image, but if that's not the best way to do it I'm open to suggestions. 我本打算使用矩形来计算图像上的汽车,但是如果这不是最好的方法,那么我可以提出建议。

Thanks. 谢谢。

using namespace cv;
using namespace std;

int main()
{
    VideoCapture TestVideo;                 //Declare video capture
    Mat frame;                              //declare Mat as frame to grab

    TestVideo.open("FroggerHighway.mp4");           //open the test video from the project directory

    if (!TestVideo.isOpened())              //If its not open declare the error
        {
            cout << "Video did not open." << endl;
            waitKey(0);
        }

    if (TestVideo.get(CV_CAP_PROP_FRAME_COUNT) < 1)     //If the frame count is less than 1, basically an error checker
        {
            cout << "Video file must have at least one frame." << endl;
            waitKey(0);
        }

    TestVideo.read(frame);                              //read the first frame
    Mat frameGray = Mat::zeros(frame.size(), CV_8UC1);  //Convert frame source to gray
    cvtColor(frame, frameGray, CV_BGR2GRAY); 

    Mat frame2 = Mat::zeros(frameGray.size(), frameGray.type());    //Intermediate frame
    Mat framediff;                                                  //Frame differencing
    Mat thresh;
    Mat element;                                                    //Element used for morphOps (dilation)
    Mat dil;    

    while (TestVideo.isOpened() & waitKey(30) != 27)                //while the video is open, show the frame, press escape to end video
        {
        absdiff(frameGray, frame2, framediff);                      //take absolute difference of both frames
        threshold(framediff, thresh, 22, 255, CV_THRESH_BINARY);    //If absdiff is greater than 22, turn it white. 

        namedWindow("Gray", CV_WINDOW_NORMAL);                      //Display gray video
        imshow("Gray", frameGray);

        namedWindow("FrameDiff", CV_WINDOW_NORMAL);                 //Show frame difference before threshold/dilation
        imshow("FrameDiff", framediff);

        namedWindow("Threshold", CV_WINDOW_NORMAL);                 //Show thresholded video
        imshow("Threshold", thresh);

        element = getStructuringElement(MORPH_CROSS,                //morphOps dilation
            Size(2 * 5 + 1, 2 * 5 + 1),
            Point(5, 5));
        dilate(thresh, dil, element, Point(-1, -1), 1, 1, 1);

        namedWindow("Dilation", CV_WINDOW_NORMAL);                  //Show dilated video. 
        imshow("Dilation", dil);

        //Apply findCountours function to draw countours and count the objects.
        vector<vector<Point> > contours;        //Not sure what this does but it works
        findContours(dil, contours, RETR_EXTERNAL, CHAIN_APPROX_SIMPLE);    //(outout image, hierarchy, and 2 ways to calculate it)
        vector<vector<Point> > contours_poly(contours.size());      //Also not sure what this does
        vector<Rect> boundRect(contours.size());            //This is used to approximate a polygon to fit the contours it calculated I think

        Mat output = Mat::zeros(dil.rows, dil.cols, CV_8UC3);   
        int counter = 0;                                    //Used to count # of rectangle drawn

        for (int i = 0; i < contours.size(); i++)
        {
            approxPolyDP(Mat(contours[i]), contours_poly[i], 3, true);  //Approximates a polygon to fit the contours calculated ?
            boundRect[i] = boundingRect(Mat(contours_poly[i]));    //for each approximation, a bounding rectangle is sorted around the contour ?

            if ((boundRect[i].x * boundRect[i].y) > 40000)          //If the bounding rectangle has an area less than 40,000 then just ignore it completely
            {
                counter = counter + 1;
                drawContours(output, contours, i, Scalar(255, 255, 255), -3);   //(input, countors, contour to be drawn, color of it, thickness (negative fills image));
                rectangle(output, boundRect[i].tl(), boundRect[i].br(), Scalar(0, 0, 255), 2, 8, 0); //Draws the actual rectangle around the contours
            }
        }

        cout << "Rectangles Drawn: " << counter << endl;

        namedWindow("Output", CV_WINDOW_NORMAL);
        imshow("Output", output);

            if (((TestVideo.get(CV_CAP_PROP_POS_FRAMES) + 1) < TestVideo.get(CV_CAP_PROP_FRAME_COUNT)) & (waitKey(30) != 27))   //Move the frame count up 1, show the frame
            {
                TestVideo.read(frame);
                frameGray.copyTo(frame2);                   //MUST USE copyTo, or clone! Can't do frame2 = frameGray*
                cvtColor(frame, frameGray, CV_BGR2GRAY);
            }
            else
            {
                cout << "End of Video" << endl;
                waitKey(0);
                break;
            }

            waitKey(30);        //wait 30ms between showing each frame
        }

    return (0);
}

You're multiplying the x and y coordinates to get the area of rectangle, you should be multiplying the width and height . 您将xy坐标相乘以获得矩形的面积,应该将widthheight相乘。

//If the bounding rectangle has an area less than 40,000 then just ignore it completely
if ((boundRect[i].width * boundRect[i].height) > 40000)          
{
// code here
}

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM