/*
Mat bwareaopen(Mat& img, int size)
{
CBlobResult blobs;
blobs = CBlobResult( img ,Mat(),4);
blobs.Filter( blobs, B_INCLUDE, CBlobGetLength(), B_GREATER, size );
Mat newimg(img.size(),img.type());
newimg.setTo(0);
for(int i=0;i<blobs.GetNumBlobs();i++)
{
blobs.GetBlob(i)->FillBlob(newimg,CV_RGB(255,255,255),0,0,true);
}
return newimg;
}
Mat removeUseless(Mat& img, int low, int hight)
{
CBlobResult blobs;
blobs = CBlobResult( img ,Mat(),4);
blobs.Filter( blobs, B_OUTSIDE, CBlobGetLength(), low, hight );
Mat newimg(img.size(),img.type());
newimg.setTo(0);
for(int i=0;i<blobs.GetNumBlobs();i++)
{
blobs.GetBlob(i)->FillBlob(newimg,CV_RGB(255,255,255),0,0,true);
}
return newimg;
}
*/
Mat computeWhiteMaskLight(Mat& input){
Mat img, gray;
input.clone().convertTo(img,CV_64F);
Mat BGRbands[3] = {Mat::zeros(img.size(), CV_64F), Mat::zeros(img.size(), CV_64F),Mat::zeros(img.size(), CV_64F)};
Mat I1 = Mat::zeros(img.size(), CV_64F);
Mat I2 = Mat::zeros(img.size(), CV_64F);
Mat I3 = Mat::zeros(img.size(), CV_64F);
Mat Id1 = Mat::zeros(img.size(), CV_64F);
Mat Id2 = Mat::zeros(img.size(), CV_64F);
Mat Id3 = Mat::zeros(img.size(), CV_64F);
Mat I = Mat::zeros(img.size(), CV_64F);
Mat mask = Mat::zeros(img.size(), CV_8U);
vector< vector<Point> > contours;
Mat mask2 = Mat::zeros(img.size(), CV_8U);
split(img,BGRbands);
I1 = BGRbands[0]-BGRbands[1];
I1 = I1.mul(I1);
I2 = BGRbands[0]-BGRbands[2];
I2 = I2.mul(I2);
I3 = BGRbands[1]-BGRbands[2];
I3 = I3.mul(I3);
Id1 = I1-I2;
Id2 = I1-I3;
Id3 = I3-I2;
Id1 = Id1.mul(Id1);
Id2 = Id2.mul(Id2);
Id3 = Id3.mul(Id3);
I = Id1+Id2+Id3;
sqrt(I,I);
sqrt(I,I);
I.convertTo(mask,CV_8U);
threshold(mask,mask,50,255,THRESH_BINARY_INV);
Mat d = detectShadows(input);
bitwise_and(mask,d,mask);
cvtColor(input,gray,CV_BGR2GRAY);
threshold(gray,mask2,160,255,THRESH_BINARY);
mask = mask+mask2;
medianBlur(mask,mask,9);
findContours(mask, contours, CV_RETR_CCOMP, CV_CHAIN_APPROX_SIMPLE);
vector<double> areas = computeArea(contours);
for(int j = areas.size()-1; j>=0; j--){
if(areas.at(j)>MAX_AREA || areas.at(j)<MIN_AREA ) // 10000 400
contours.erase(contours.begin()+j);
}
Mat out = Mat::zeros(Size(img.cols,img.rows), CV_8U);
for (int idx = 0; idx < contours.size(); idx++)
drawContours(out, contours, idx, Scalar(255,255,255), CV_FILLED, 8);
return out;
}
int main()
{
// Opening video and testing integrity ---------------------------/
VideoCapture capture(FILENAME);
if (!capture.isOpened())
{
cerr << "Nao conseguiu abrir o video.\n";
return -1;
}
//----------------------------------------------------------------/
for (int i = 0; i < samples.size(); i++)
{
samples[i] = Mat(Size(WIDTH, HEIGHT), CV_8UC3);
}
while (cont_n < SAMPLES)
{
bool bSuccess = capture.read(src);
if (!bSuccess)
{
cout << "ERROR: could not read frame from file..." << endl;
break;
}
img_8u3c = resizeFixed(src);
if (cont_t % INTERVAL == 0)
{
for (int j = 0; j < HEIGHT; j++)
{
for (int i = 0; i < WIDTH; i++)
{
samples[cont_n].at<Vec3b>(j, i) = img_8u3c.at<Vec3b>(j, i);
}
}
cont_n++;
}
cont_t++;
}
for (int j = 0; j < HEIGHT; j++)
{
for (int i = 0; i < WIDTH; i++)
{
for (int c = 0; c < SAMPLES; c++)
{
pixel_list[c] = samples[c].at<Vec3b>(j, i);
}
sort(begin(pixel_list), end(pixel_list), Compare_Vec3f);
bg_8u3c.at<Vec3b>(j, i) = pixel_list[SAMPLES / 2 + 1];
}
}
// At this point we have an aproximation of the background
// This is used to start the samples in the Vibe algorithm
// imshow("Bg gerado", bg_8u3c);
cvtColor(bg_8u3c, bg_lab_8u3c, CV_BGR2Lab);
// Background gradients will be used in the texture patch test
bg_8u3c.convertTo(bg_32fc3, CV_32FC3, 1 / 255.0);
calcGradients(bg_32fc3, bg_dx_32f, bg_dy_32f, bg_mag_32f, bg_ori_32f);
initBackground(bg_32fc3);
Mat vibe_mask(Size(WIDTH, HEIGHT), CV_8U);
Mat vibe_filtered(Size(WIDTH, HEIGHT), CV_8U);
// Start of the real-time analysis (after initial bg generation) --------------------------/
while (true)
{
bool bSuccess = capture.read(src);
if (!bSuccess)
{
cout << "ERROR: could not read frame from file..." << endl;
return -1;
}
img_8u3c = resizeFixed(src);
// In this point we have the Mat image holding
// a smaller version of the actual frame.
cvtColor(img_8u3c, img_lab_8u3c, CV_BGR2Lab);
img_8u3c.convertTo(img_32fc3, CV_32FC3, 1 / 255.0);
// Bloco para gerar mascara - usado no lugar do Vibe para debug -----------------------/
if (!VIBE)
{
Mat img_8u_gray, bg_8u_gray;
cvtColor(img_8u3c, img_8u_gray, CV_BGR2GRAY);
cvtColor(bg_8u3c, bg_8u_gray, CV_BGR2GRAY);
absdiff(img_8u_gray, bg_8u_gray, diff);
threshold(diff, mask_8u, 50, 10, CV_8U);
morphologyEx(mask_8u, mask_8u, MORPH_CLOSE, getStructuringElement(MORPH_ELLIPSE, Size(5, 5)));
//GaussianBlur(mask_8u, mask_8u, Size(3, 3), 0);
filtered_mask_8u = media_binary(mask_8u, 3, 10);
}
else
{
// Vibe -------------------------------------------------------------------------------/
vibe(img_32fc3, vibe_mask);
morphologyEx(vibe_mask, vibe_mask, MORPH_CLOSE, getStructuringElement(MORPH_ELLIPSE, Size(5, 5)));
filtered_mask_8u = media_binary(vibe_mask, 3, 10);
}
// ------------------------------------------------------------------------------------/
//findConnectedComponents(filtered_mask_8u, components);
//components.clear();
findConnectedComponents(filtered_mask_8u, components);
// For all connected components:
for (int i = 0; i < components.size(); i++)
{
Rect roi = components[i];
calcGradients(Mat(img_32fc3, roi), Mat(img_dx_32f, roi),
Mat(img_dy_32f, roi), Mat(img_mag_32f, roi), Mat(img_ori_32f, roi));
/*imshow("ori_bg", bg_ori_32f);
imshow("ori_fg", img_ori_32f);
Mat diff_ori_ffs;
absdiff(img_ori_32f, bg_ori_32f, diff_ori_ffs);
threshold(diff_ori_ffs, diff_ori_ffs, 0.75, 255, CV_8U);
imshow("ori_diff", Mat(diff_ori_ffs, roi));
waitKey(1000000);*/
// Detect shadows
detectShadows(Mat(img_8u3c, roi), Mat(img_lab_8u3c, roi), Mat(bg_lab_8u3c, roi), Mat(filtered_mask_8u, roi),
Mat(img_dx_32f, roi), Mat(img_dy_32f, roi), Mat(img_mag_32f, roi), Mat(img_ori_32f, roi),
Mat(bg_dx_32f, roi), Mat(bg_dy_32f, roi), Mat(bg_mag_32f, roi), Mat(img_ori_32f, roi), roi);
}
components.clear();
//imshow("mask", filtered_mask_8u);
//imshow("vibe", vibe_filtered);
//imshow("img", img_8u3c);
imshow("final", img_8u3c);
waitKey(10000000);
switch (waitKey(1)) {
case ESC_KEY:
return 0;
}
}
capture.release();
return 0;
}
