-
Notifications
You must be signed in to change notification settings - Fork 1
/
main.cpp
56 lines (47 loc) · 1.7 KB
/
main.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
// Copyright (C) 2011 NICTA (www.nicta.com.au)
// Copyright (C) 2011 Andres Sanin
//
// This file is provided without any warranty of fitness for any purpose.
// You can redistribute this file and/or modify it under the terms of
// the GNU General Public License (GPL) as published by the
// Free Software Foundation, either version 3 of the License
// or (at your option) any later version.
// (see https://www.opensource.org/licenses for more info)
#include <opencv2/imgproc/imgproc.hpp>
#include <opencv2/highgui/highgui.hpp>
#include "ChromacityShadRem.h"
#include "GeometryShadRem.h"
#include "LrTextureShadRem.h"
#include "PhysicalShadRem.h"
#include "SrTextureShadRem.h"
int main() {
// load frame, background and foreground
cv::Mat frame = cv::imread("samples/frame.bmp");
cv::Mat bg = cv::imread("samples/bg.bmp");
cv::Mat fg = cv::imread("samples/fg.bmp", CV_LOAD_IMAGE_GRAYSCALE);
// create shadow removers
ChromacityShadRem chr;
PhysicalShadRem phy;
GeometryShadRem geo;
SrTextureShadRem srTex;
LrTextureShadRem lrTex;
// matrices to store the masks after shadow removal
cv::Mat chrMask, phyMask, geoMask, srTexMask, lrTexMask;
// remove shadows
chr.removeShadows(frame, fg, bg, chrMask);
phy.removeShadows(frame, fg, bg, phyMask);
geo.removeShadows(frame, fg, bg, geoMask);
srTex.removeShadows(frame, fg, bg, srTexMask);
lrTex.removeShadows(frame, fg, bg, lrTexMask);
// show results
cv::imshow("frame", frame);
cv::imshow("bg", bg);
cv::imshow("fg", fg);
cv::imshow("chr", chrMask);
cv::imshow("phy", phyMask);
cv::imshow("geo", geoMask);
cv::imshow("srTex", srTexMask);
cv::imshow("lrTex", lrTexMask);
cv::waitKey();
return 0;
}