2012-12-11 23 views
4

我想測試背景減法的ViBe算法。目前我正在使用opencv庫。我在opencv/samples/gpu/bgfg_segm.cpp和bgfg_vibe.cpp文件中找到了一個示例實現。這些文件在gpu模塊下。現在我有一個沒有GPU的系統。當我嘗試運行代碼時,它在第一幀的初始化時崩潰。任何人都可以告訴我如何解決這個問題?是否可以使用ViBe算法,在opencv中實現,用於沒有GPU的系統?

在此先感謝。

回答

7

僞代碼吸大的時間!這是非僞/版本。結果:有隻嘗試還是儘量不要

bgfg_vibe.hpp

#ifndef bgfg_vibe_hpp 
#define bgfg_vibe_hpp 
#include "opencv2/core/core.hpp" 
#include "opencv2/imgproc/imgproc.hpp" 
#include "opencv2/highgui/highgui.hpp" 

struct Model { 
    cv::Mat*** samples; 
    cv::Mat** fgch; 
    cv::Mat* fg; 
}; 

class bgfg_vibe 
{ 
#define rndSize 256 
    unsigned char ri; 
#define rdx ri++ 
public: 
    bgfg_vibe(); 
    int N,R,noMin,phi; 
    void init_model(cv::Mat& firstSample); 
    void setphi(int phi); 
    cv::Mat* fg(cv::Mat& frame); 
private: 
    bool initDone; 
    cv::RNG rnd; 
    Model* model; 
    void init(); 
    void fg1ch(cv::Mat& frame,cv::Mat** samples,cv::Mat* fg); 
    int rndp[rndSize],rndn[rndSize],rnd8[rndSize]; 
}; 

#endif 

bgfg_vibe.cpp

#include "bgfg_vibe.hpp" 

bgfg_vibe::bgfg_vibe():R(20),N(20),noMin(2),phi(0) 
{ 
    initDone=false; 
    rnd=cv::theRNG(); 
    ri=0; 
} 
void bgfg_vibe::init() 
{ 
    for(int i=0;i<rndSize;i++) 
    { 
     rndp[i]=rnd(phi); 
     rndn[i]=rnd(N); 
     rnd8[i]=rnd(8); 
    } 
} 
void bgfg_vibe::setphi(int phi) 
{ 
    this->phi=phi; 
    for(int i=0;i<rndSize;i++) 
    { 
     rndp[i]=rnd(phi); 
    } 
} 
void bgfg_vibe::init_model(cv::Mat& firstSample) 
{ 
    std::vector<cv::Mat> channels; 
    split(firstSample,channels); 
    if(!initDone) 
    { 
     init(); 
     initDone=true; 
    } 
    model=new Model; 
    model->fgch= new cv::Mat*[channels.size()]; 
    model->samples=new cv::Mat**[N]; 
    model->fg=new cv::Mat(cv::Size(firstSample.cols,firstSample.rows), CV_8UC1); 
    for(size_t s=0;s<channels.size();s++) 
    {  
     model->fgch[s]=new cv::Mat(cv::Size(firstSample.cols,firstSample.rows), CV_8UC1); 
     cv::Mat** samples= new cv::Mat*[N]; 
     for(int i=0;i<N;i++) 
     { 
      samples[i]= new cv::Mat(cv::Size(firstSample.cols,firstSample.rows), CV_8UC1); 
     } 
     for(int i=0;i<channels[s].rows;i++) 
     { 
      int ioff=channels[s].step.p[0]*i; 
      for(int j=0;j<channels[0].cols;j++) 
      { 
       for(int k=0;k<1;k++) 
       { 
        (samples[k]->data + ioff)[j]=channels[s].at<uchar>(i,j); 
       } 
       (model->fgch[s]->data + ioff)[j]=0; 

       if(s==0)(model->fg->data + ioff)[j]=0; 
      } 
     } 
     model->samples[s]=samples; 
    } 
} 
void bgfg_vibe::fg1ch(cv::Mat& frame,cv::Mat** samples,cv::Mat* fg) 
{ 
    int step=frame.step.p[0]; 
    for(int i=1;i<frame.rows-1;i++) 
    { 
     int ioff= step*i; 
     for(int j=1;j<frame.cols-1;j++) 
     { 
      int count =0,index=0; 
      while((count<noMin) && (index<N)) 
      { 
       int dist= (samples[index]->data + ioff)[j]-(frame.data + ioff)[j]; 
       if(dist<=R && dist>=-R) 
       { 
        count++; 
       } 
       index++; 
      } 
      if(count>=noMin) 
      { 
       ((fg->data + ioff))[j]=0; 
       int rand= rndp[rdx]; 
       if(rand==0) 
       { 
        rand= rndn[rdx]; 
        (samples[rand]->data + ioff)[j]=(frame.data + ioff)[j]; 
       } 
       rand= rndp[rdx]; 
       int nxoff=ioff; 
       if(rand==0) 
       { 
        int nx=i,ny=j; 
        int cases= rnd8[rdx]; 
        switch(cases) 
        { 
        case 0: 
         //nx--; 
         nxoff=ioff-step; 
         ny--; 
         break; 
        case 1: 
         //nx--; 
         nxoff=ioff-step; 
         ny; 
         break; 
        case 2: 
         //nx--; 
         nxoff=ioff-step; 
         ny++; 
         break; 
        case 3: 
         //nx++; 
         nxoff=ioff+step; 
         ny--; 
         break; 
        case 4: 
         //nx++; 
         nxoff=ioff+step; 
         ny; 
         break; 
        case 5: 
         //nx++; 
         nxoff=ioff+step; 
         ny++; 
         break; 
        case 6: 
         //nx; 
         ny--; 
         break; 
        case 7: 
         //nx; 
         ny++; 
         break; 
        } 
        rand= rndn[rdx]; 
        (samples[rand]->data + nxoff)[ny]=(frame.data + ioff)[j]; 
       } 
      }else 
      { 
       ((fg->data + ioff))[j]=255; 
      } 
     } 
    } 
} 
cv::Mat* bgfg_vibe::fg(cv::Mat& frame) 
{ 
    std::vector<cv::Mat> channels; 
    split(frame,channels); 
    for(size_t i=0;i<channels.size();i++) 
    { 
     fg1ch(channels[i],model->samples[i],model->fgch[i]);   
     if(i>0 && i<2) 
     { 
      bitwise_or(*model->fgch[i-1],*model->fgch[i],*model->fg); 
     } 
     if(i>=2) 
     { 
      bitwise_or(*model->fg,*model->fgch[i],*model->fg); 
     } 
    } 
    if(channels.size()==1) return model->fgch[0]; 
    return model->fg; 
} 

的main.cpp

#include "bgfg_vibe.hpp" 
using namespace cv; 

int main(int argc, char ** argv) 
{ 
    Mat frame; 
    bgfg_vibe bgfg; 
    VideoCapture cap("c:/toprocess/frame_%04d.jpg"); 
    cap >> frame; 
    bgfg.init_model(frame); 
    for(;;) 
    { 
     cap>>frame; 
     Mat fg = *bgfg.fg(frame); 
     imshow("fg",fg); 
     waitKey(1); 
    } 
    return 0; 
} 
+1

感謝您的回覆。它已經這麼久了。那麼,我還沒有測試ViBe的上述實現,但它似乎非常有幫助。至少,上面的代碼是ViBe的非gpu版本,代表了紙張的僞代碼。所以,有興趣的人可以試試這個。 – sam18

+0

非常感謝您的代碼@Zaw Lin。它完美的作品。但是,我注意到了一些我不確定的事情。在bgfg_vibe.cpp中應該是這一行: model-> samples = new cv :: Mat ** [N]; be: model-> samples = new cv :: Mat ** [3]; //用於3聲道圖像。 – tnq177

+1

嗨,N是正確的。對於多通道圖像,每個通道都被視爲灰色,實現只是爲每個通道初始化單個模型。 –

0

我和你有同樣的問題:想在OpenCV中使用沒有GPU的Vibe。不幸的是,在當前版本的OpenCV(2.4.3)中沒有非gpu實現。如果我錯了,請告訴我。所以我基於論文中提供的僞代碼實施了自己的氛圍。這並不像我想的那麼困難。

到目前爲止我對此算法感到滿意。總的來說這很好,但我認爲還需要一些進一步的工作來克服幽靈問題。

+0

感謝您的回覆。順便說一句你的經驗與僞代碼?我的意思是,你能告訴我一些統計數據或者其他什麼,比如分辨率和fps?我已經實現了MOG,對於640 X 480分辨率幀,我將達到30幀/秒,學習參數設置爲0.您認爲PBAS能夠以自己的方式解決鬼影問題嗎? – sam18

0

請注意,盛傳是專利而且它的使用受許可費的限制。請前往vibeinmotion.com獲取試用版和更多信息。我負責ViBE的業務開發,您可以直接聯繫我獲取更多信息。問候。

+0

具有完整SQL異常的網站不存在任何信息。 – Horizon1710

相關問題