-
Notifications
You must be signed in to change notification settings - Fork 5
/
BackGroundSubstractor.java
65 lines (51 loc) · 1.82 KB
/
BackGroundSubstractor.java
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
57
58
59
60
61
62
63
64
65
package org.genericsystem.cv;
import javax.swing.ImageIcon;
import javax.swing.JFrame;
import javax.swing.JLabel;
import org.genericsystem.cv.utils.NativeLibraryLoader;
import org.opencv.core.Core;
import org.opencv.core.Mat;
import org.opencv.video.BackgroundSubtractorMOG2;
import org.opencv.video.Video;
import org.opencv.videoio.VideoCapture;
public class BackGroundSubstractor {
static {
NativeLibraryLoader.load();
}
private static VideoCapture camera = new VideoCapture(0);
private final static String adjustedDirectoryPath2 = "aligned-image-3.png";
// private static List<Mat> getClassMats2() {
// return Arrays.stream(new File(adjustedDirectoryPath2).listFiles()).filter(img -> img.getName().endsWith(".png")).map(img -> Imgcodecs.imread(img.getPath())).collect(Collectors.toList());
// }
public static void main(String[] args) {
JFrame jframe = new JFrame("BackgroundSubstractor");
jframe.setResizable(false);
jframe.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
JLabel vidpanel = new JLabel();
jframe.setContentPane(vidpanel);
// List<Mat> srcs = getClassMats2();
Mat img = new Mat();
// srcs.get(0).copyTo(img);
camera.read(img);
jframe.setSize(img.width(), img.height());
jframe.setVisible(true);
BackgroundSubtractorMOG2 mog2 = Video.createBackgroundSubtractorMOG2();
// BackgroundSubtractorKNN knn = Video.createBackgroundSubtractorKNN();
Mat average = null;
for (;;) {
// srcs.get(i % srcs.size()).copyTo(img);
camera.read(img);
Mat fgMask = new Mat();
mog2.apply(img, fgMask);
Mat fg = new Mat();
img.copyTo(fg, fgMask);
if (average == null)
average = fg;
else
Core.addWeighted(average, 1 - 1d / 30, fg, 1d / 30, 0, average);
ImageIcon image = new ImageIcon(Tools.mat2bufferedImage(average));
vidpanel.setIcon(image);
vidpanel.repaint();
}
}
}