OpenCV-Head-Detection/CDetection.cpp at master · alimsahy/OpenCV-Head-Detection · GitHub

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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
//
// Created by Alimsah YILDIRIM on 5/1/19.
//

#include "CDetection.h"

CDetection::CDetection(const string &windowName)
{
	this->m_windowName = windowName;
	namedWindow(this->m_windowName);
}

void CDetection::processFrame()
{
	VideoCapture capture(0);
	capture.set(CV_CAP_PROP_FPS, 30);
	while (true)
	{
		capture >> this->m_frame;

		resize(this->m_frame, this->m_frame, Size(1000, 700), this->m_scalingFactor, this->m_scalingFactor, INTER_AREA);

		cvtColor(this->m_frame, this->m_frameGray, COLOR_BGR2GRAY);
		equalizeHist(this->m_frameGray, this->m_frameGray);
		this->m_classifier.detectMultiScale(this->m_frameGray, this->m_faces, 1.2, 9, 0 | CV_HAAR_SCALE_IMAGE, Size(30, 30));

		for (auto &face: this->m_faces)
		{
			this->m_detectedFaceCount = this->m_faces.size();

			Rect faceRect(face.x, face.y + 10, face.width, face.height);

			int x = face.x - int(0.0 * face.width);
			int y = face.y - int(0.0 * face.height);
			int w = int(1.0 * face.width);
			int h = int(1.0 * face.height);

			this->m_frameROI = this->m_frame(Rect(x, y, w, h));
			resize(this->m_faceMask, this->m_faceMaskSmall, Size(w, h));
			cvtColor(this->m_faceMaskSmall, this->m_grayMaskSmall, COLOR_BGR2GRAY);
			threshold(this->m_grayMaskSmall, this->m_grayMaskSmallThresh, 230, 255, THRESH_BINARY_INV);
			bitwise_not(this->m_grayMaskSmallThresh, this->m_grayMaskSmallThreshInv);
			bitwise_and(this->m_faceMaskSmall, this->m_faceMaskSmall, this->m_maskedFace, this->m_grayMaskSmallThresh);
			bitwise_and(this->m_frameROI, this->m_frameROI, this->m_maskedFrame, this->m_grayMaskSmallThreshInv);

			if (x > 0 && y > 0 && x + w < this->m_frame.cols && y + h < this->m_frame.rows)
			{
				add(this->m_maskedFace, this->m_maskedFrame, this->m_frame(Rect(x, y, w, h)));
			}
		}
		imshow(this->m_windowName, this->m_frame);
		waitKey(1);
	}
}

void CDetection::processFrame(VideoCapture &capture)
{
	while (true)
	{
		capture >> this->m_frame;

		resize(this->m_frame, this->m_frame, Size(1000, 700), this->m_scalingFactor, this->m_scalingFactor, INTER_AREA);

		cvtColor(this->m_frame, this->m_frameGray, COLOR_BGR2GRAY);
		equalizeHist(this->m_frameGray, this->m_frameGray);
		this->m_classifier.detectMultiScale(this->m_frameGray, this->m_faces, 1.2, 9, 0 | CV_HAAR_SCALE_IMAGE, Size(30, 30));

		for (auto &face: this->m_faces)
		{
			Rect faceRect(face.x, face.y + 10, face.width, face.height);

			int x = face.x - int(0.0 * face.width);
			int y = face.y - int(0.0 * face.height);
			int w = int(1.0 * face.width);
			int h = int(1.0 * face.height);

			this->m_frameROI = this->m_frame(Rect(x, y, w, h));
			resize(this->m_faceMask, this->m_faceMaskSmall, Size(w, h));
			cvtColor(this->m_faceMaskSmall, this->m_grayMaskSmall, COLOR_BGR2GRAY);
			threshold(this->m_grayMaskSmall, this->m_grayMaskSmallThresh, 230, 255, THRESH_BINARY_INV);
			bitwise_not(this->m_grayMaskSmallThresh, this->m_grayMaskSmallThreshInv);
			bitwise_and(this->m_faceMaskSmall, this->m_faceMaskSmall, this->m_maskedFace, this->m_grayMaskSmallThresh);
			bitwise_and(this->m_frameROI, this->m_frameROI, this->m_maskedFrame, this->m_grayMaskSmallThreshInv);

			if (x > 0 && y > 0 && x + w < this->m_frame.cols && y + h < this->m_frame.rows)
			{
				add(this->m_maskedFace, this->m_maskedFrame, this->m_frame(Rect(x, y, w, h)));
			}
		}
		imshow(this->m_windowName, this->m_frame);
		waitKey(1);
	}
}

void CDetection::setSource(int source, const string &sourceAddress)
{
	switch (source)
	{

	}
}

void CDetection::setClassifier(int classifier)
{
	switch (classifier)
	{
		case CDetection::ClassifierType::FrontalFace:
		{
			this->m_classifier.load("haarcascade_frontalface_default.xml");
			break;
		}
		case CDetection::ClassifierType ::FrontalFace2:
		{
			this->m_classifier.load("haarcascade_frontalface_alt2.xml");
			break;
		}
		default:
		{
			this->m_classifier.load("haarcascade_frontalface.xml");
			break;
		}
	}
}

void CDetection::setFacemask(const string &mask)
{
	this->m_faceMask = imread(mask);
}

void CDetection::setScalingFactor(const float &factor)
{

}