void in_received_handler(DictionaryIterator *iter, void *context)
{
(void) context;
vibe();
//Get data
Tuple *t = dict_read_first(iter);
while(t != NULL)
{
process_tuple(t);
//Get next
t = dict_read_next(iter);
}
}
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;
}
