我想測試背景減法的ViBe算法。目前我正在使用opencv庫。我在opencv/samples/gpu/bgfg_segm.cpp和bgfg_vibe.cpp文件中找到了一個示例實現。這些文件在gpu模塊下。現在我有一個沒有GPU的系統。當我嘗試運行代碼時,它在第一幀的初始化時崩潰。任何人都可以告訴我如何解決這個問題?是否可以使用ViBe算法,在opencv中實現,用於沒有GPU的系統?
在此先感謝。
我想測試背景減法的ViBe算法。目前我正在使用opencv庫。我在opencv/samples/gpu/bgfg_segm.cpp和bgfg_vibe.cpp文件中找到了一個示例實現。這些文件在gpu模塊下。現在我有一個沒有GPU的系統。當我嘗試運行代碼時,它在第一幀的初始化時崩潰。任何人都可以告訴我如何解決這個問題?是否可以使用ViBe算法,在opencv中實現,用於沒有GPU的系統?
在此先感謝。
僞代碼吸大的時間!這是非僞/版本。結果:有隻嘗試還是儘量不要
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;
}
我和你有同樣的問題:想在OpenCV中使用沒有GPU的Vibe。不幸的是,在當前版本的OpenCV(2.4.3)中沒有非gpu實現。如果我錯了,請告訴我。所以我基於論文中提供的僞代碼實施了自己的氛圍。這並不像我想的那麼困難。
到目前爲止我對此算法感到滿意。總的來說這很好,但我認爲還需要一些進一步的工作來克服幽靈問題。
感謝您的回覆。順便說一句你的經驗與僞代碼?我的意思是,你能告訴我一些統計數據或者其他什麼,比如分辨率和fps?我已經實現了MOG,對於640 X 480分辨率幀,我將達到30幀/秒,學習參數設置爲0.您認爲PBAS能夠以自己的方式解決鬼影問題嗎? – sam18
請注意,盛傳是專利而且它的使用受許可費的限制。請前往vibeinmotion.com獲取試用版和更多信息。我負責ViBE的業務開發,您可以直接聯繫我獲取更多信息。問候。
具有完整SQL異常的網站不存在任何信息。 – Horizon1710
感謝您的回覆。它已經這麼久了。那麼,我還沒有測試ViBe的上述實現,但它似乎非常有幫助。至少,上面的代碼是ViBe的非gpu版本,代表了紙張的僞代碼。所以,有興趣的人可以試試這個。 – sam18
非常感謝您的代碼@Zaw Lin。它完美的作品。但是,我注意到了一些我不確定的事情。在bgfg_vibe.cpp中應該是這一行: model-> samples = new cv :: Mat ** [N]; be: model-> samples = new cv :: Mat ** [3]; //用於3聲道圖像。 – tnq177
嗨,N是正確的。對於多通道圖像,每個通道都被視爲灰色,實現只是爲每個通道初始化單個模型。 –