opencv cvblob - 渲染除白色之外的blob

时间:2011-08-21 18:54:43

标签: opencv object-detection

我正在尝试使用cvblob检测对象。不知何故,我的代码只标记白色对象。如何标记其他颜色的物体,如一罐啤酒或一瓶水。

这是我的代码:

#include "highgui.h"
#include "cv.h"
#include "cvaux.h"
#include "iostream"
#include <stdio.h>
#include <ctype.h>

#include <cvblob.h>

using namespace cv;
using namespace std;
using namespace cvb;

int main(int argc, char** argv) {
    CvTracks tracks;

    cvNamedWindow("frame", CV_WINDOW_AUTOSIZE);
    cvMoveWindow("frame", 50, 100);

    CvCapture* capture;

    IplImage* frame = 0;

    capture = cvCreateCameraCapture( 0 ); //capture frames from cam on index 0: /dev/video0/

    if (!capture) {
        return -1;
    }

    cvSetCaptureProperty(capture, CV_CAP_PROP_FRAME_WIDTH, 320);
    cvSetCaptureProperty(capture, CV_CAP_PROP_FRAME_HEIGHT, 240);

    frame = cvQueryFrame(capture);

    while(frame) {
        IplImage *gray = cvCreateImage(cvGetSize(frame), IPL_DEPTH_8U, 1);
        cvCvtColor(frame, gray, CV_BGR2GRAY);
        cvThreshold(gray, gray, 150, 255, CV_THRESH_BINARY);

        IplImage *labelImg=cvCreateImage(cvGetSize(gray), IPL_DEPTH_LABEL, 1);
        CvBlobs blobs;
        unsigned int result=cvLabel(gray, labelImg, blobs);

        cvFilterByArea(blobs, 500, 1000000);
//        cvRenderBlobs(labelImg, blobs, frame, frame, CV_BLOB_RENDER_BOUNDING_BOX);
        cvRenderBlobs(labelImg, blobs, frame, frame, CV_BLOB_RENDER_CENTROID);
        cvUpdateTracks(blobs, tracks, 200., 5);
        cvRenderTracks(tracks, frame, frame, CV_TRACK_RENDER_ID|CV_TRACK_RENDER_BOUNDING_BOX);


        for (CvBlobs::const_iterator it=blobs.begin(); it!=blobs.end(); ++it) {
            cout << "Blob #" << it->second->label << ": Area=" << it->second->area << ", Centroid=(" << it->second->centroid.x << ", " << it->second->centroid.y << ")" << endl;
        }

        cvShowImage("frame", frame);

        cout << "----------------------------" << endl;

        frame = cvQueryFrame(capture);

        char c = cvWaitKey(10);
        if(c==27)
            break;

    }
}

任何提示都表示赞赏。

米洛

2 个答案:

答案 0 :(得分:4)

默认情况下这是选项,如果不更改cvblob库中的源代码,则无法更改它。

如果你真的想要改变这很容易,你可以创建一个相同方法的副本,添加一个新的输入变量如CvScalar来选择输出颜色。这很容易。

方法cvRenderBlob将在cvcontour.cpp。

我已经在cvblob库中做了很多改进,并且在接下来的几个月里我会把它推给创建者。

答案 1 :(得分:0)

尝试添加:

“cvInRangeS(hsvframe,cvScalar(23,41,133),cvScalar(40,150,255),threshy); // for yellow”

在过滤blob之前。它是一系列HSV(而非RGB)值,用于定义所需颜色的阈值。

希望它有所帮助。